BeyzaTopbas commited on
Commit
428b8ca
·
verified ·
1 Parent(s): 859cede

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
- model = joblib.load("digit_model.pkl")
 
 
 
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