Update app.py

app.py

@@ -14,8 +14,7 @@ model = tf.keras.Sequential([
 ])
 
 # Compile the model with the Adam optimizer and loss function
-optimizer = tf.keras.optimizers.Adam(0.01)
-model.compile(optimizer=optimizer, loss='mse')
+model.compile(optimizer=tf.keras.optimizers.Adam(0.1), loss='mean_squared_error')
 
 # Training data (Celsius to Fahrenheit)
 celsius = np.array([-40, -10, 0, 8, 15, 22, 38], dtype=float)
@@ -28,7 +27,7 @@ input_celsius = st.number_input('Enter Celsius value:', value=0.0, format="%.1f"
 if st.button('Train Model and Predict Fahrenheit'):
     with st.spinner('Training...'):
         # Fit the model
-        model.fit(celsius, fahrenheit, epochs=100)
+        history = model.fit(celsius, fahrenheit, epochs=500)
     st.success('Training completed!')
 
     # Make prediction
@@ -47,3 +46,9 @@ if st.button('Train Model and Predict Fahrenheit'):
     plt.ylabel('Fahrenheit')
     plt.legend()
     st.pyplot(plt)
+
+    plt.xlabel('Epoch Number')
+    plt.ylabel("Loss Magnitude")
+    plt.plot(history.history['loss'])
+    plt.legend()
+    st.pyplot(plt)