Upload folder using huggingface_hub

app.py

@@ -3,7 +3,10 @@ import joblib
 import pandas as pd
 from flask import Flask, request, jsonify
 
+# Initialize Flask app
 app = Flask(__name__)
+
+# Load the model
 model = joblib.load('Extraalearn.joblib')
 
 EXPECTED_FEATURES = [
@@ -18,7 +21,7 @@ EXPECTED_FEATURES = [
 
 @app.get('/')
 def home():
-    return 'ExtraaLearn API is Running'
+    return 'ExtraaLearn Lead Conversion API is Running'
 
 @app.post('/v1/predict')
 def predict():
@@ -41,9 +44,10 @@ def predict():
         p['digital_media_Yes'] = 1 if data.get('digital_media') == 'Yes' else 0
         p['educational_channels_Yes'] = 1 if data.get('educational_channels') == 'Yes' else 0
         p['referral_Yes'] = 1 if data.get('referral') == 'Yes' else 0
+
         df_final = pd.DataFrame([p])[EXPECTED_FEATURES]
-        prediction_proba = model.predict_proba(df_final)[:, 1].tolist()[0] # Use predict_proba
-        return jsonify({'Conversion_Probability': float(prediction_proba)}) # Changed key to Conversion_Probability
+        prediction_proba = model.predict_proba(df_final)[:, 1].tolist()[0]
+        return jsonify({'Conversion_Probability': float(prediction_proba)})
     except Exception as e:
         return jsonify({'error': str(e)}), 400