Update app.py

app.py

@@ -20,39 +20,16 @@ def predict_with_loaded_model(in_sentences):
 # Streamlit interface
 st.title("Stress Prediction with DistilBERT")
 
-# Initialize session state variables for user input and prediction output
-if 'user_input' not in st.session_state:
-    st.session_state.user_input = ""
-if 'prediction' not in st.session_state:
-    st.session_state.prediction = None
-
 # Add a text input box for the user to enter a sentence
-user_input = st.text_area("Enter a sentence or text:", st.session_state.user_input)
-
-# Define buttons with a sidebar layout for easy spacing
-col1, col2 = st.columns([1, 1])
-with col1:
-    # When the user clicks "Predict", run the prediction function
-    if st.button("Predict"):
-        if user_input:
-            # Update session state with the user input
-            st.session_state.user_input = user_input
-            # Make the prediction using the model
-            st.session_state.prediction = predict_with_loaded_model([user_input])[0]
-        else:
-            st.write("Please enter a sentence to predict.")
-
-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()  # Refresh the app
-
-# Display the prediction result
-if st.session_state.prediction:
-    prediction = st.session_state.prediction
-    st.write(f"Text: {prediction['text']}")
-    st.write(f"Prediction: {prediction['label']}")
-    # st.write(f"Confidence: {prediction['confidence']}")
+user_input = st.text_area("Enter a sentence or text:", "")
+
+# When the user clicks "Predict", run the prediction function
+if st.button("Predict"):
+    if user_input:
+        # Make the prediction using the model
+        prediction = predict_with_loaded_model([user_input])[0]
+        st.write(f"Text: {prediction['text']}")
+        st.write(f"Prediction: {prediction['label']}")
+        # st.write(f"Confidence: {prediction['confidence']}")
+    else:
+        st.write("Please enter a sentence to predict.")