BodduSriPavan111 commited on
Commit
53aa976
·
verified ·
1 Parent(s): 63abb4d

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +7 -7
src/streamlit_app.py CHANGED
@@ -34,12 +34,12 @@ st.markdown(page_bg, unsafe_allow_html=True)
34
  # ---------------- Load Model + Preprocessors ----------------
35
  @st.cache_resource
36
  def load_model_and_preprocessors():
37
- model = lgb.Booster(model_file="fast_lightgbm.txt")
38
- scaler = joblib.load("scaler.pkl")
39
- rbf = joblib.load("rbf.pkl")
40
- num_cols = joblib.load("num_cols.pkl")
41
- cat_cols = joblib.load("cat_cols.pkl")
42
- cat_mappings = joblib.load("cat_mappings.pkl")
43
  return model, scaler, rbf, num_cols, cat_cols, cat_mappings
44
 
45
  model, scaler, rbf, num_cols, cat_cols, cat_mappings = load_model_and_preprocessors()
@@ -95,7 +95,7 @@ row_df = pd.DataFrame([user_input])
95
  from PIL import Image
96
 
97
  # Load logo/image
98
- logo = Image.open("assets/kitiki.png") # <-- place your png file in the same folder
99
 
100
  col1, col2 = st.columns([1, 4]) # logo left, text right
101
  with col1:
 
34
  # ---------------- Load Model + Preprocessors ----------------
35
  @st.cache_resource
36
  def load_model_and_preprocessors():
37
+ model = lgb.Booster(model_file="src/fast_lightgbm.txt")
38
+ scaler = joblib.load("src/scaler.pkl")
39
+ rbf = joblib.load("src/rbf.pkl")
40
+ num_cols = joblib.load("src/num_cols.pkl")
41
+ cat_cols = joblib.load("src/cat_cols.pkl")
42
+ cat_mappings = joblib.load("src/cat_mappings.pkl")
43
  return model, scaler, rbf, num_cols, cat_cols, cat_mappings
44
 
45
  model, scaler, rbf, num_cols, cat_cols, cat_mappings = load_model_and_preprocessors()
 
95
  from PIL import Image
96
 
97
  # Load logo/image
98
+ logo = Image.open("src/kitiki.png") # <-- place your png file in the same folder
99
 
100
  col1, col2 = st.columns([1, 4]) # logo left, text right
101
  with col1: