Upload 4 files

Dockerfile

@@ -0,0 +1,21 @@
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the requirements file into the container
+COPY requirements.txt .
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application code into the container
+COPY . .
+
+# Expose the port the app runs on
+EXPOSE 5000
+
+EXPOSE 7860
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 CMD curl -f http://localhost:7860/ || exit 1
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,45 @@
+from flask import Flask, request, jsonify
+import joblib
+import pandas as pd
+import numpy as np
+
+# Initialize the Flask application
+app = Flask(__name__)
+
+# Load the trained model pipeline
+try:
+    model = joblib.load('superkart_model.pkl')
+    print("Model loaded successfully.")
+except Exception as e:
+    print(f"Error loading model: {e}")
+    model = None
+
+@app.route('/')
+def home():
+    return "SuperKart Sales Prediction API is running!"
+
+@app.route('/predict', methods=['POST'])
+def predict():
+    if model is None:
+        return jsonify({'error': 'Model not loaded'}), 500
+
+    try:
+        # Get data from the POST request
+        data = request.get_json(force=True)
+        
+        # Convert the incoming JSON data into a pandas DataFrame
+        # The model pipeline expects a DataFrame with the original column names
+        input_data = pd.DataFrame(data, index=[0])
+        
+        # Make a prediction
+        prediction = model.predict(input_data)
+        
+        # Return the prediction as JSON
+        return jsonify({'predicted_sales': np.round(prediction[0], 2)})
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 400
+
+if __name__ == '__main__':
+    # Run the app on port 5000
+    app.run(host='0.0.0.0', port=5000)

requirements.txt

@@ -0,0 +1,7 @@
+Flask==3.0.0
+scikit-learn==1.3.2
+pandas==2.0.3
+numpy==1.25.2
+xgboost==2.0.3
+joblib==1.3.2
+gunicorn==21.2.0

superkart_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:157d7f1c6b8e66278bbb080eed840ed33a2bb64501562b96bd02eb6d02fc7ce9
+size 31487338