Upload folder using huggingface_hub

introvert_extrovert_predictor.py

@@ -31,7 +31,7 @@ def predict_personality_single():
     # Convert to DataFrame and predict
     input_df = pd.DataFrame([sample])
     prediction = model.predict(input_df).tolist()[0]
-    personality = "Extrovert" if prediction == 1 else "Introvert"
+    personality = "Extrovert" if prediction == 0 else "Introvert"
 
     return jsonify({'Predicted_Personality': personality})