Upload folder using huggingface_hub

introvert_extrovert_predictor.py

@@ -8,6 +8,8 @@ app = Flask("Introvert Extrovert Predictor")
 # Load the trained classification model
 model = joblib.load("introvert_extrovert_predictor_v1_0.joblib")
 
+api_hit_counter = 0
+
 # Home route
 @app.get("/")
 def home():
@@ -33,6 +35,10 @@ def predict_personality_single():
     prediction = model.predict(input_df).tolist()[0]
     personality = "Extrovert" if prediction == 0 else "Introvert"
 
+    # Increment and print the counter
+    api_hit_counter += 1
+    print(f"API hit count: {api_hit_counter}")
+
     return jsonify({'Predicted_Personality': personality})
 
 @app.post("/v1/personality/predictbatch")