Update app.py

app.py

@@ -13,12 +13,19 @@ model = tf.keras.Sequential([
     tf.keras.layers.Dense(units=1, input_shape=[1])
 ])
 
-# Optimizer selection
-optimizer = st.sidebar.selectbox(
+# Optimizer selection with learning rate adjustment
+optimizer_choice = st.sidebar.selectbox(
     "Select optimizer",
     ('sgd', 'adam', 'rmsprop')
 )
 
+learning_rate = st.sidebar.slider("Select learning rate", 0.001, 0.1, 0.01)
+
+optimizer = tf.keras.optimizers.get({
+    'class_name': optimizer_choice,
+    'config': {'learning_rate': learning_rate}
+})
+
 # Compile the model with an optimizer and loss function
 model.compile(optimizer=optimizer, loss='mse')