Update app.py

app.py

@@ -107,9 +107,11 @@ def main():
     st.header("Is it human-made 📝 or Generated with AI 🤖 ?  ")
     
     # Check the model to use
-    option = st.selectbox("Select a model to use:", models_available)
-    if option and "prediction" in st.session_state:
-        del st.session_state.prediction
+    def restore_prediction_state():
+        if "prediction" in st.session_state:
+            del st.session_state.prediction
+    option = st.selectbox("Select a model to use:", models_available, on_change=restore_prediction_state)
+    
 
     # Load the selected trained model
     if option in ("BERT-based model", "DistilBERT-based model (BERT light)"):
@@ -160,7 +162,7 @@ def main():
 
     if st.button("Model Explanation"):
         # Check if there's text in the session state
-        if "text" in st.session_state:
+        if "text" in st.session_state and "prediction" in st.session_state:
            
             if option in ("Naive Bayes", "Logistic Regression"):
                  with st.spinner('Wait for it 💭...'):