superkart-frontend / src /streamlit_app.py
omm7's picture
Upload frontend Streamlit files
088ab9d
raw
history blame contribute delete
373 Bytes
import streamlit as st
import pandas as pd
import pickle
# load assets
with open("model/xgb_superkart_model.pkl", "rb") as f:
model = pickle.load(f)
X_test = pd.read_csv("data/X_test.csv")
# predict
preds = model.predict(X_test)
# output
st.title("SuperKart Sales Forecast")
st.subheader("X_test Predictions")
st.dataframe(pd.DataFrame({"Predicted Sales": preds}))