WalmartSales / app.py
VipDest's picture
Update app.py
cf88c8a verified
import gradio as gr
from prediction_script import predict # Import your prediction function
iface = gr.Interface(
fn=predict,
inputs=gr.Dataframe(
headers=["Store", "Holiday_Flag", "Temperature", "Fuel_Price", "CPI", "Unemployment",
"Weekly_Sales_Lag1", "Weekly_Sales_Lag2", "Weekly_Sales_Lag4",
"Sales_Fuel_Interaction", "Sales_CPI_Interaction", "Sales_Unemployment_Interaction",
"Month", "Quarter", "Year", "Holiday_Impact"],
type="pandas"
),
outputs=gr.Number(),
title="Walmart Sales Prediction",
description="Enter the features to predict weekly sales."
)
iface.launch()