Upload folder using huggingface_hub

app.py

@@ -26,11 +26,10 @@ def home():
     return "HI, Welcome to the Extraa Learn conversion Predictor API!"
 
 # Define an endpoint for single property prediction (POST request)
-@rental_price_predictor_api.post('/v1/conversion')
+@@rental_price_predictor_api.post('/v1/conversion')
 def predict_rental_price():
     property_data = request.get_json()
 
-    # Fix tuple bug by removing commas
     sample = {
         'age': property_data['age'],
         'website_visits': property_data['website_visits'],
@@ -41,16 +40,18 @@ def predict_rental_price():
         'profile_completed': property_data['profile_completed'],
         'last_activity': property_data['last_activity'],
         'print_media_type1': property_data['print_media_type1'],
-        'print_media_type2': property_data['print_media_type2'],  # ✅ no comma
-        'digital_media': property_data['digital_media'],          # ✅ no comma
+        'print_media_type2': property_data['print_media_type2'],
+        'digital_media': property_data['digital_media'],
         'educational_channels': property_data['educational_channels'],
         'referral': property_data['referral']
     }
 
     input_data = pd.DataFrame([sample])
-    status_log = model.predict(input_data)[0]
-    status = float(np.exp(status_log))
-    return jsonify({'Predicted Status': status})
+
+    # Directly predict class (0 or 1)
+    predicted_status = int(model.predict(input_data)[0])
+
+    return jsonify({'Predicted Status': predicted_status})
 
 
 # Define an endpoint for batch prediction (POST request)