SandeepMM commited on
Commit
ed6af1a
·
verified ·
1 Parent(s): 9dcbd74

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. Dockerfile +16 -0
  2. XGBoostRegressor_BEST_Pipeline.joblib +3 -0
  3. app.py +49 -0
  4. requirements.txt +11 -0
Dockerfile ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9-slim
2
+
3
+ # Set the working directory inside the container
4
+ WORKDIR /app
5
+
6
+ # Copy all files from the current directory to the container's working directory
7
+ COPY . .
8
+
9
+ # Install dependencies from the requirements file without using cache to reduce image size
10
+ RUN pip install --no-cache-dir --upgrade -r requirements.txt
11
+
12
+ # Define the command to start the application using Gunicorn with 4 worker processes
13
+ # - `-w 4`: Uses 4 worker processes for handling requests
14
+ # - `-b 0.0.0.0:7860`: Binds the server to port 7860 on all network interfaces
15
+ # - `app:app`: Runs the Flask app (assuming `app.py` contains the Flask instance named `app`)
16
+ CMD ["gunicorn", "-w", "4", "-b", "0.0.0.0:7860", "app:app"]
XGBoostRegressor_BEST_Pipeline.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1316e0f1ffa836f435afbed414f60cb1ca4f0d1ab02c9b3c3a3c3246ef4aabb2
3
+ size 2236826
app.py ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import joblib
2
+ import pandas as pd
3
+ from flask import Flask, request, jsonify
4
+
5
+ # Initialize Flask app with a name
6
+ app = Flask("SuperKart Sales Predictor")
7
+
8
+ # Load the trained churn prediction model
9
+ model = joblib.load("RandomForestRegressor_Model_BEST.joblib")
10
+
11
+ # Define a route for the home page
12
+ @app.get('/')
13
+ def home():
14
+ return "Welcome to the SuperKart Sales Prediction API"
15
+
16
+ # Define an endpoint to predict churn for a single customer
17
+ @app.post('/v1/product')
18
+ def predict_churn():
19
+ # Get JSON data from the request
20
+ customer_data = request.get_json()
21
+
22
+ # Extract relevant customer features from the input data
23
+ sample = {
24
+ 'Product_Id': customer_data['Product_Id']
25
+ 'Product_Weight': customer_data['Product_Weight'],
26
+ 'Product_Sugar_Content': customer_data['Product_Sugar_Content'],
27
+ 'Product_Allocated_Area': customer_data['Product_Allocated_Area'],
28
+ 'Product_Type': customer_data['Product_Type'],
29
+ 'Product_MRP': customer_data['Product_MRP'],
30
+ 'Store_Id': customer_data['Store_Id'],
31
+ 'Store_Establishment_Year': customer_data['Store_Establishment_Year'],
32
+ 'Store_Size': customer_data['Store_Size'],
33
+ 'Store_Location_City_Type': customer_data['Store_Location_City_Type'],
34
+ 'Store_Type': customer_data['Store_Type']
35
+ }
36
+
37
+ # Convert the extracted data into a DataFrame
38
+ input_data = pd.DataFrame([sample])
39
+
40
+ # Make a Sales prediction using the trained model
41
+ prediction = model.predict(input_data).tolist()[0]
42
+
43
+ # Return the prediction as a JSON response
44
+ return jsonify({'Prediction': prediction})
45
+
46
+
47
+ # Run the Flask app in debug mode
48
+ if __name__ == '__main__':
49
+ app.run(debug=True)
requirements.txt ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ pandas==2.2.2
2
+ numpy==2.0.2
3
+ scikit-learn==1.6.1
4
+ xgboost==2.1.4
5
+ joblib==1.4.2
6
+ Werkzeug==2.2.2
7
+ flask==2.2.2
8
+ gunicorn==20.1.0
9
+ requests==2.28.1
10
+ uvicorn[standard]
11
+ streamlit==1.43.2