Marc-Adrien commited on
Commit
fa338e2
·
verified ·
1 Parent(s): 83a104b

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. Dockerfile +16 -0
  2. app.py +61 -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,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 Prediction")
7
+
8
+ # Section for online prediction
9
+ st.subheader("Online Prediction")
10
+
11
+
12
+ # Collect user input for property features
13
+ product_weight = st.number_input("Product weight", min_value=4.0, max_value=22.0, step=0.5)
14
+ product_allocated_area = st.number_input("product_allocated_area", min_value=0.004, max_value=0.3, step=0.001)
15
+ product_mrp = st.number_input("product_allocated_area", min_value=31, max_value=266, step=1)
16
+ Store_Establishment_Year = st.number_input("Store_Establishment_Year", min_value=1987, max_value=2009, step=1)
17
+ Product_Sugar_Content = st.selectbox("Product_Sugar_Content", ["Low Sugar", "Regular", "No Sugar", "reg"])
18
+ Product_Type = st.selectbox("Product_Type", ["Fruits and Vegetables", "Snack Foods","Frozen Foods","Dairy","Household","Baking Goods",
19
+ "Canned","Health and Hygiene","Meat","Soft Drinks","Breads","Hard Drinks","Others","Starchy Foods","Breakfast","Seafood"])
20
+ Store_Id = st.selectbox("Store_Id", ["OUT004", "OUT001", "OUT002", "OUT003"])
21
+ Store_Location_City_Type = st.selectbox("Store_Location_City_Type", ["Tier 2", "Tier 1", "Tier 3"])
22
+ Store_Type = st.selectbox("Store_Type", ["Supermarket Type2", "Supermarket Type1", "Departmental Store","Food Mart"])
23
+
24
+ # Convert user input into a DataFrame
25
+ input_data = pd.DataFrame([{
26
+ 'Product_weight': product_weight,
27
+ 'product_allocated_area': product_allocated_area,
28
+ 'product_mrp': product_mrp,
29
+ 'Store_Establishment_Year': Store_Establishment_Year,
30
+ 'Product_Sugar_Content': Product_Sugar_Content,
31
+ 'Product_Type': Product_Type,
32
+ 'Store_Id': Store_Id,
33
+ 'Store_Location_City_Type': Store_Location_City_Type,
34
+ 'Store_Type': Store_Type
35
+ }])
36
+
37
+ # Make prediction when the "Predict" button is clicked
38
+ if st.button("Predict"):
39
+ response = requests.post("https://<username>-<repo_id>.hf.space/v1/sales", json=input_data.to_dict(orient='records')[0]) # Send data to Flask API
40
+ if response.status_code == 200:
41
+ prediction = response.json()['Predicted Sales)']
42
+ st.success(f"Predicted sales: {prediction}")
43
+ else:
44
+ st.error("Error making prediction.")
45
+
46
+ # Section for batch prediction
47
+ st.subheader("Batch Prediction")
48
+
49
+ # Allow users to upload a CSV file for batch prediction
50
+ uploaded_file = st.file_uploader("Upload CSV file for batch prediction", type=["csv"])
51
+
52
+ # Make batch prediction when the "Predict Batch" button is clicked
53
+ if uploaded_file is not None:
54
+ if st.button("Predict Batch"):
55
+ response = requests.post("https://<username>-<repo_id>.hf.space/v1/salesbatch", files={"file": uploaded_file}) # Send file to Flask API
56
+ if response.status_code == 200:
57
+ predictions = response.json()
58
+ st.success("Batch predictions completed!")
59
+ st.write(predictions) # Display the predictions
60
+ else:
61
+ 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