Update src/streamlit_app.py
Browse files- src/streamlit_app.py +4 -1
src/streamlit_app.py
CHANGED
|
@@ -11,7 +11,10 @@ st.write(os.listdir())
|
|
| 11 |
# ======================
|
| 12 |
# LOAD MODEL
|
| 13 |
# ======================
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
# ======================
|
| 17 |
# PAGE CONFIG
|
|
|
|
| 11 |
# ======================
|
| 12 |
# LOAD MODEL
|
| 13 |
# ======================
|
| 14 |
+
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
| 15 |
+
model_path = os.path.join(BASE_DIR, "digit_model.pkl")
|
| 16 |
+
|
| 17 |
+
model = joblib.load(model_path)
|
| 18 |
|
| 19 |
# ======================
|
| 20 |
# PAGE CONFIG
|