BeyzaTopbas commited on
Commit
a686aff
·
verified ·
1 Parent(s): d334d84

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +5 -2
src/streamlit_app.py CHANGED
@@ -11,15 +11,18 @@ st.set_page_config(
11
  st.title("Facial Keypoints Detection")
12
  st.write("Upload a face image and the model will predict facial keypoints.")
13
 
14
- MODEL_PATH = "facial_keypoints_resnet.h5"
 
 
15
 
16
  @st.cache_resource
17
  def load_model():
18
  return tf.keras.models.load_model(
19
  MODEL_PATH,
20
- safe_mode=False
21
  )
22
 
 
23
  model = load_model()
24
 
25
  uploaded_file = st.file_uploader(
 
11
  st.title("Facial Keypoints Detection")
12
  st.write("Upload a face image and the model will predict facial keypoints.")
13
 
14
+ import os
15
+
16
+ MODEL_PATH = os.path.join(os.path.dirname(__file__), "facial_keypoints_resnet.h5")
17
 
18
  @st.cache_resource
19
  def load_model():
20
  return tf.keras.models.load_model(
21
  MODEL_PATH,
22
+ safe_mode=False
23
  )
24
 
25
+
26
  model = load_model()
27
 
28
  uploaded_file = st.file_uploader(