Upload folder using huggingface_hub

app.py

@@ -20,17 +20,23 @@ def home():
 @app.route("/predict", methods=["POST"])
 def predict():
     try:
-        # Get JSON input
         data = request.get_json(force=True)
-        features = np.array(data["features"]).reshape(1, -1)
+        features = np.array(data["features"])
 
-        # Predict
-        prediction = model.predict(features)[0]
+        # Case 1: single row
+        if features.ndim == 1:
+            features = features.reshape(1, -1)
+            prediction = model.predict(features)[0]
+            return jsonify({"predicted_revenue": float(prediction)})
 
-        return jsonify({"predicted_revenue": float(prediction)})
+        # Case 2: multiple rows
+        else:
+            predictions = model.predict(features).tolist()
+            return jsonify({"predictions": predictions})
 
     except Exception as e:
         return jsonify({"error": str(e)})
 
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)