Update app_pages/page1_model_comparison.py

app_pages/page1_model_comparison.py

@@ -11,11 +11,25 @@ def run():
     
     # Example placeholder for user input
     user_input = st.text_area("")
-    
+
+
+    if st.button('Classify with All Models'):
+        # Bag of Words/TF-IDF prediction
+        bow_tfidf_result = predict(user_input)
+        st.write(f'{class_prefix} {bow_tfidf_result} according to Bag of Words/TF-IDF.')
+        
+        # LSTM prediction
+        lstm_result = predict_review(user_input)
+        st.write(f'{class_prefix} {lstm_result} according to LSTM.')
+        
+        # ruBERT prediction
+        rubert_result = classify_text(user_input)
+        st.write(f'{class_prefix} {rubert_result} according to ruBERT.')
+
     # Placeholder buttons for model selection
-    if st.button('Classify with BoW/TF-IDF'):
-        st.write(f'{class_prefix}{predict(user_input)}')
-    if st.button('Classify with LSTM'):
-        st.write(f'{class_prefix}{predict_review(user_input)}')
-    if st.button('Classify with ruBERT'):
-        st.write(f'{class_prefix}{classify_text(user_input)}')
+    # if st.button('Classify with BoW/TF-IDF'):
+    #     st.write(f'{class_prefix}{predict(user_input)}')
+    # if st.button('Classify with LSTM'):
+    #     st.write(f'{class_prefix}{predict_review(user_input)}')
+    # if st.button('Classify with ruBERT'):
+    #     st.write(f'{class_prefix}{classify_text(user_input)}')