Update app.py

app.py

@@ -178,29 +178,3 @@ with col1:
 with col2:
     #st.text_area("Metrics:", value=metrics, height=100, disabled=True)
     st.json(st.session_state.metrics)
-
-
-# Show RMSE Button
-col3, col4 = st.columns([3, 4])  # For RMSE button
-
-with col3:
-    if st.button("Show RMSE"):     
-        if st.session_state.metrics:  # Check if metrics exist
-            # Assuming compute_rmse returns a numeric value or dictionary
-            rmse_result = compute_rmse(list(st.session_state.metrics.values()), st.session_state.response)
-            
-            # Check the type of the RMSE result and ensure it's a valid JSON object
-            if isinstance(rmse_result, (int, float)):  # If the result is a numeric value
-                st.session_state.rmse = {"rmse_value": rmse_result}
-            elif isinstance(rmse_result, dict):  # If the result is already a dictionary
-                st.session_state.rmse = rmse_result
-            else:
-                st.session_state.rmse = {"error": "Invalid RMSE result type"}
-        else:
-            st.session_state.rmse = {"error": "Metrics not available"}
-
-# Display RMSE (whether metrics were calculated or not)
-with col4:
-    # Make sure to pass a valid JSON object (dictionary or list)
-    if isinstance(st.session_state.rmse, dict):
-        st.json(st.session_state.rmse)  # This should now work correctly