TokenTutor's picture
Upload folder using huggingface_hub
9ef4a35 verified
import streamlit as st
import pandas as pd
import requests
API_ENDPOINT="https://TokenTutor-ProductSalesRevenuePrediction.hf.space/v1/product_sales_revenue"
#product type
product_types = [
"Fruits and Vegetables",
"Snack Foods",
"Frozen Foods",
"Dairy",
"Household",
"Baking Goods",
"Canned",
"Health and Hygiene",
"Meat",
"Soft Drinks",
"Breads",
"Hard Drinks",
"Others",
"Starchy Foods",
"Breakfast",
"Seafood"
]
#store types
store_types = [
"Food Mart",
"Supermarket Type1",
"Supermarket Type2",
"Departmental Store"
]
#Store Id
store_ids = [
"OUT001",
"OUT002",
"OUT003",
"OUT004"
]
store_Location_City_Types=[
"Tier 1",
"Tier 2",
"Tier 3"
]
store_sizes=[
"Small",
"Medium",
"Large"
]
#Set title of the Streamlit app
st.title("Product Revenue prediction")
#Section for online prediction
st.subheader("Online Prediction")
#Collect user input for features
# Product_Weight = st.number_input("Product Weight", min_value=4.0, max_value=25.0, step=0.5)
# Product_Sugar_Content = st.selectbox("Product Sugar Content", ["No Sugar", "Low Sugar", "Regular"])
# Product_Allocated_Area = st.number_input("Product Allocated Area", min_value=0.001, max_value=0.3)
# Product_Type = st.selectbox("Product Type", product_types)
# Product_MRP = st.number_input("Product MRP", min_value=30.0, max_value=300.0)
# Store_Id = st.selectbox("Store Id", store_ids)
# Store_Establishment_Year = st.number_input("Store Establishment Year", min_value=1988, max_value=2010, step=1)
# Store_Size = st.selectbox("Store Size", store_sizes)
# Store_Location_City_Type = st.selectbox("Store Location City Type", store_Location_City_Types)
# Store_Type = st.selectbox("Store Type", store_types)
# payload = {
# 'Product_Weight': Product_Weight,
# 'Product_Sugar_Content': Product_Sugar_Content,
# 'Product_Allocated_Area': Product_Allocated_Area,
# 'Product_Type': Product_Type ,
# 'Product_MRP': Product_MRP,
# 'Store_Id': Store_Id,
# 'Store_Establishment_Year': Store_Establishment_Year,
# 'Store_Size': Store_Size,
# 'Store_Location_City_Type': Store_Location_City_Type,
# 'Store_Type': Store_Type
# }
# if st.button("Predict"):
# response = requests.post(API_ENDPOINT, json=payload)
# if response.status_code == 200:
# json_data= response.json()
# st.write('Predicted Sales revenue ', json_data.get('Prediction'))
# else:
# st.write(f"Error making prediction: {response.status_code}")