Maalai commited on
Commit
60aa834
·
verified ·
1 Parent(s): 5401f05

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. Dockerfile +9 -13
  2. app.py +64 -0
  3. requirements.txt +3 -3
Dockerfile CHANGED
@@ -1,20 +1,16 @@
1
- FROM python:3.13.5-slim
 
2
 
 
3
  WORKDIR /app
4
 
5
- RUN apt-get update && apt-get install -y \
6
- build-essential \
7
- curl \
8
- git \
9
- && rm -rf /var/lib/apt/lists/*
10
-
11
- COPY requirements.txt ./
12
- COPY src/ ./src/
13
 
 
14
  RUN pip3 install -r requirements.txt
15
 
16
- EXPOSE 8501
17
-
18
- HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
19
 
20
- ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
1
+ # Use a minimal base image with Python 3.9 installed
2
+ FROM python:3.9-slim
3
 
4
+ # Set the working directory inside the container to /app
5
  WORKDIR /app
6
 
7
+ # Copy all files from the current directory on the host to the container's /app directory
8
+ COPY . .
 
 
 
 
 
 
9
 
10
+ # Install Python dependencies listed in requirements.txt
11
  RUN pip3 install -r requirements.txt
12
 
13
+ # Define the command to run the Streamlit app on port 8501 and make it accessible externally
14
+ CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false"]
 
15
 
16
+ # NOTE: Disable XSRF protection for easier external access in order to make batch predictions
app.py ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ import requests
4
+
5
+ # Set the title of the Streamlit app
6
+ st.title("Superkart product sales Predictor")
7
+
8
+ # Section for Single Product Prediction
9
+ st.subheader("Single Product Prediction")
10
+
11
+
12
+ # Collect user input for product features
13
+ Product_Weight = st.number_input("Product Weight", min_value=1.0)
14
+ Product_Sugar_Content = st.selectbox("Sugar content in the product", ['Low Sugar', 'Regular', 'No Sugar', 'reg'])
15
+ Product_Allocated_Area = st.number_input("Allocated area for the product")
16
+ Product_Type = st.selectbox("Broad category for each product", ['Frozen Foods', 'Dairy', 'Canned', 'Baking Goods',
17
+ 'Health and Hygiene', 'Snack Foods', 'Meat', 'Household',
18
+ 'Hard Drinks', 'Fruits and Vegetables', 'Breads', 'Soft Drinks',
19
+ 'Breakfast', 'Others', 'Starchy Foods', 'Seafood'])
20
+ Product_MRP = st.number_input("Maximum retail price of each product", min_value=1.0)
21
+ Store_Establishment_Year = st.number_input("Year in which the store was established", min_value=1980, step=1, value=1987)
22
+ Store_Size = st.selectbox("Size of the store, depending on sq. feet", ['Medium', 'High', 'Small'])
23
+ Store_Location_City_Type = st.selectbox("Type of city in which the store is located", ['Tier 2', 'Tier 1', 'Tier 3'])
24
+ Store_Type = st.selectbox("Type of store depending on the products that are being sold there", ['Supermarket Type2', 'Departmental Store', 'Supermarket Type1',
25
+ 'Food Mart'])
26
+
27
+ # Convert user input into a DataFrame
28
+ input_data = pd.DataFrame([{
29
+ 'productWeight': Product_Weight,
30
+ 'productSugarContent': Product_Sugar_Content,
31
+ 'productAllocatedArea': Product_Allocated_Area,
32
+ 'prouctType': Product_Type,
33
+ 'productMRP': Product_MRP,
34
+ 'storeEstablishmentYear': Store_Establishment_Year,
35
+ 'storeSize': Store_Size,
36
+ 'storeLocationCityType': Store_Location_City_Type,
37
+ 'storeType': Store_Type
38
+ }])
39
+
40
+ # Make prediction when the "Predict" button is clicked
41
+ if st.button("Predict"):
42
+ response = requests.post("https://Maalai-SuperKartSalesPredictionBackend.hf.space/v1/products", json=input_data.to_dict(orient='records')[0]) # Send data to Flask API
43
+ if response.status_code == 200:
44
+ prediction = response.json()['Predicted Price (in dollars)']
45
+ st.success(f"Predicted Product store sales total: {prediction}")
46
+ else:
47
+ st.error("Error making prediction.")
48
+
49
+ # Section for batch prediction
50
+ st.subheader("Batch Prediction")
51
+
52
+ # Allow users to upload a CSV file for batch prediction
53
+ uploaded_file = st.file_uploader("Upload CSV file for batch prediction", type=["csv"])
54
+
55
+ # Make batch prediction when the "Predict Batch" button is clicked
56
+ if uploaded_file is not None:
57
+ if st.button("Predict Batch"):
58
+ response = requests.post("https://Maalai-SuperKartSalesPredictionBackend.hf.space/v1/productsbatch", files={"file": uploaded_file}) # Send file to Flask API
59
+ if response.status_code == 200:
60
+ predictions = response.json()
61
+ st.success("Batch predictions completed!")
62
+ st.write(predictions) # Display the predictions
63
+ else:
64
+ st.error("Error making batch prediction.")
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
- altair
2
- pandas
3
- streamlit
 
1
+ pandas==2.2.2
2
+ requests==2.28.1
3
+ streamlit==1.43.2