Update app.py

app.py

@@ -330,7 +330,7 @@ with tab1:
     # Handle restart button click
     if restart_button:
         st.session_state.restart_clicked = True
-        st.experimental_rerun()
+        st.rerun()  # Using st.rerun() instead of st.experimental_rerun()
     
     # Display prediction
     if predict_button: