Upload folder using huggingface_hub

introvert_extrovert_predictor.py

@@ -49,7 +49,7 @@ def predict_personality_batch():
 
     # Predict personality
     predictions = model.predict(input_df).tolist()
-    input_df["Predicted_Personality"] = ["Extrovert" if p == 1 else "Introvert" for p in predictions]
+    input_df["Predicted_Personality"] = ["Extrovert" if p == 0 else "Introvert" for p in predictions]
 
     # Convert to dict for JSON output
     result = input_df.to_dict(orient="records")