Upload folder using huggingface_hub

app.py

@@ -0,0 +1,23 @@
+import flask
+from flask import Flask, request, jsonify
+import joblib
+import pandas as pd
+import numpy as np
+
+app = Flask(__name__)
+
+saved_model_path = "/app/superkart_model_v1_0.joblib" # Path inside Docker container
+loaded_model = joblib.load(saved_model_path)
+
+@app.route('/predict', methods=['POST'])
+def predict():
+    try:
+        json_ = request.json
+        query_df = pd.DataFrame(json_)
+        predictions = loaded_model.predict(query_df)
+        return jsonify({'predictions': predictions.tolist()})
+    except Exception as e:
+        return jsonify({'error': str(e)}), 400
+
+if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0', port=5000)