Update app.py

app.py

@@ -45,9 +45,10 @@ with col1:
 with col2:
     # Refresh button to reset the user input and output
     if st.button("Refresh"):
+        # Clear both user input and prediction output
         st.session_state.user_input = ""
         st.session_state.prediction = None
-        st.rerun()  # Use st.rerun() instead of st.experimental_rerun()
+        st.rerun()  # Refresh the app
 
 # Display the prediction result
 if st.session_state.prediction:
@@ -55,4 +56,3 @@ if st.session_state.prediction:
     st.write(f"Text: {prediction['text']}")
     st.write(f"Prediction: {prediction['label']}")
     # st.write(f"Confidence: {prediction['confidence']}")
-