Upload folder using huggingface_hub

app.py

@@ -14,7 +14,8 @@ def home():
     return "Welcome to the Forecasting Model API!"
 
 # Single prediction endpoint
-@app.post('/v1/params')
+#@app.post('/v1/params')
+@app.route('/v1/params', methods=['POST'])
 def predict_sales():
     # Get JSON data from the request
     input_dict = request.get_json()
@@ -43,7 +44,8 @@ def predict_sales():
     return jsonify({'Prediction': prediction})
 
 # Batch prediction endpoint
-@app.post('/v1/paramsbatch')
+#@app.post('/v1/paramsbatch')
+@app.route('/v1/paramsbatch', methods=['POST'])
 def predict_sales_batch():
     # Get uploaded CSV file
     file = request.files['file']

requirements.txt

@@ -9,3 +9,4 @@ gunicorn==20.1.0
 requests==2.28.1
 uvicorn[standard]
 streamlit==1.43.2
+flask>=2.0