Update app.py

app.py

@@ -60,13 +60,22 @@ if st.button("Predict Personality by Test"):
     data['Predicted'] = data.apply(get_predictions, axis=1)
     st.dataframe(data[['True_label','MAX1','MAX2','MAX3', 'Predicted']])
 
-if st.button("Predict Personality by Open Question"):
+if st.button("Predict Personality by Open Question F"):
     def get_predictions(row):
         processed_text = row['processed_text_open']
         result = classifier(processed_text, labels)
         highest_score_label = result['labels'][0]  # Assumes the labels are sorted by score, highest first
         return highest_score_label
         
+    
+
+    # Apply predictions across all rows
+    data_open['Predicted_F'] = data_open.apply(get_predictionsM, axis=1)
+    st.dataframe(data_open[['True_label', 'Predicted_F']])
+
+if st.button("Predict Personality by Open Question M"):
+    
+        
     def get_predictionsM(row):
         processed_text = row['processed_text_mopen']
         result = classifier(processed_text, labels)
@@ -75,5 +84,4 @@ if st.button("Predict Personality by Open Question"):
 
     # Apply predictions across all rows
     data_open['Predicted_M'] = data_open.apply(get_predictions, axis=1)
-    data_open['Predicted_F'] = data_open.apply(get_predictionsM, axis=1)
-    st.dataframe(data_open[['True_label', 'Predicted_F', 'Predicted_M']])
+    st.dataframe(data_open[['True_label', 'Predicted_M']])