pragmat commited on
Commit
e2b00c4
·
verified ·
1 Parent(s): 68912f1

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. Dockerfile +9 -13
  2. app.py +64 -0
  3. requirements.txt +2 -2
  4. runtime.txt +1 -0
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("Sales Revenue Prediction")
7
+
8
+ # Section for online prediction
9
+ st.subheader("Online Prediction")
10
+
11
+ # Collect user input for property features
12
+ Product_Id = st.text_input("Product Id")
13
+ Product_Weight = st.number_input("Product Weight", min_value=0.0)
14
+ Product_Sugar_Content = st.selectbox("Product Sugar Content", ["Low Sugar", "No Sugar", "Regular"])
15
+ Product_Allocated_Area = st.number_input("Product Allocated Area", min_value=0.0)
16
+ Product_Type = st.selectbox("Product Type", ["Fruits and Vegetables", "Snack Foods","Frozen Foods","Dairy",
17
+ "Household","Baking Goods","Canned","Health and Hygiene",
18
+ "Meat","Soft Drinks","Breads","Hard Drinks","Others",
19
+ "Starchy Foods","Breakfast","Seafood"])
20
+ Product_MRP = st.number_input("Product MRP", min_value=0.0)
21
+ Store_Id = st.text_input("Store Id")
22
+ Store_Establishment_Year = st.number_input("Store Establishment Year", min_value=0)
23
+ Store_Size = st.selectbox("Store Size", ["Small", "Medium", "High"])
24
+ Store_Location_City_Type = st.selectbox("Store Location City Type", ["Tier 1", "Tier 2", "Tier 3"])
25
+ Store_Type = st.selectbox("Store Type", ["Supermarket Type2", "Supermarket Type1", "Departmental Store","Food Mart"])
26
+
27
+
28
+ # Convert user input into a DataFrame
29
+ input_data = pd.DataFrame([{'Product_Id': Product_Id,
30
+ 'Product_Weight': Product_Weight,
31
+ 'Product_Sugar_Content': Product_Sugar_Content,
32
+ 'Product_Allocated_Area': Product_Allocated_Area,
33
+ 'Product_Type': Product_Type,
34
+ 'Product_MRP': Product_MRP,
35
+ 'Store_Id': Store_Id,
36
+ 'Store_Establishment_Year': Store_Establishment_Year,
37
+ 'Store_Location_City_Type': Store_Location_City_Type,
38
+ 'Store_Type': Store_Type}])
39
+
40
+ # Make prediction when the "Predict" button is clicked
41
+ if st.button("Predict"):
42
+ response = requests.post("https://pragmat-SalesRevenuePredictionBackend.hf.space/v1/revenue", json=input_data.to_dict(orient='records')[0]) # Send data to Flask API
43
+ if response.status_code == 200:
44
+ prediction = response.json()['Predicted Sales Revenue (in dollars)']
45
+ st.success(f"Predicted Sales Revenue (in dollars): {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://pragmat-SalesRevenuePredictionBackend.hf.space/v1/revenuebatch", 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
+ requests
3
+ streamlit
runtime.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ python-3.9