Upload folder using huggingface_hub

Dockerfile

@@ -1,17 +1,11 @@
-# Use an official lightweight Python image
 FROM python:3.10-slim
 
-# Set working directory
 WORKDIR /app
 
-# Copy all files into the container
 COPY . .
 
-# Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose the port Streamlit/Flask will run on
 EXPOSE 7860
 
-# Run the Flask app
-CMD ["python", "app.py"]
+CMD ["gunicorn", "-w", "4", "-b", "0.0.0.0:7860", "app:superkart_api"]

app.py

@@ -4,17 +4,17 @@ import numpy as np
 import pandas as pd
 from flask import Flask, request, jsonify
 
-app = Flask(__name__)
+superkart_api = Flask(__name__)
 
 print("🔄 Loading model...")
 model = joblib.load("superkart_model.joblib")
 print("✅ Model loaded.")
 
-@app.route("/")
+@superkart_api.route("/")
 def index():
     return "✅ SuperKart API is live."
 
-@app.route("/predict", methods=["POST"])
+@superkart_api.route("/predict", methods=["POST"])
 def predict():
     try:
         data = request.get_json(force=True)
@@ -45,4 +45,4 @@ def predict():
 
 if __name__ == "__main__":
     print("🚀 Starting Flask server...")
-    app.run(host="0.0.0.0", port=7860)
+    superkart_api.run(host="0.0.0.0", port=7860)

requirements.txt

@@ -1,4 +1,5 @@
 flask
+gunicorn
 joblib
 numpy
 pandas