pragmat commited on
Commit
dcb6150
·
verified ·
1 Parent(s): 24e8376

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. Dockerfile +16 -0
  2. app.py +64 -0
  3. requirements.txt +3 -0
Dockerfile ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ pandas==2.2.2
2
+ requests==2.28.1
3
+ streamlit==1.43.2