Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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("
|
| 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:
|