Upload folder using huggingface_hub

introvert_extrovert_predictor.py

@@ -24,8 +24,8 @@ def predict_personality_single():
         "Going_outside": input_data["Going_outside"],
         "Friends_circle_size": input_data["Friends_circle_size"],
         "Post_frequency": input_data["Post_frequency"],
-        "Stage_fear": 1 if input_data["Stage_fear"].lower() == "yes" else 0,
-        "Drained_after_socializing": 1 if input_data["Drained_after_socializing"].lower() == "yes" else 0
+        "Stage_fear": input_data["Stage_fear"],
+        "Drained_after_socializing": input_data["Drained_after_socializing"]
     }
 
     # Convert to DataFrame and predict