BeyzaTopbas commited on
Commit
b3e62bb
·
verified ·
1 Parent(s): 66b97d2

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +5 -1
src/streamlit_app.py CHANGED
@@ -13,7 +13,10 @@ st.write("Upload a face image and the model will predict facial keypoints.")
13
 
14
  import os
15
 
16
- MODEL_PATH = "facial_keypoints_resnet (1).h5"
 
 
 
17
 
18
  @st.cache_resource
19
  def load_model():
@@ -23,6 +26,7 @@ def load_model():
23
  )
24
 
25
 
 
26
  model = load_model()
27
 
28
  uploaded_file = st.file_uploader(
 
13
 
14
  import os
15
 
16
+ MODEL_PATH = os.path.join(
17
+ os.path.dirname(__file__),
18
+ "facial_keypoints_resnet.h5"
19
+ )
20
 
21
  @st.cache_resource
22
  def load_model():
 
26
  )
27
 
28
 
29
+
30
  model = load_model()
31
 
32
  uploaded_file = st.file_uploader(