BeyzaTopbas commited on
Commit
134dced
·
verified ·
1 Parent(s): 6d3af10

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +5 -0
src/streamlit_app.py CHANGED
@@ -2,12 +2,17 @@ import streamlit as st
2
  import numpy as np
3
  import tensorflow as tf
4
  from PIL import Image
 
 
 
 
5
 
6
  st.set_page_config(
7
  page_title="Facial Keypoints Detection",
8
  layout="centered"
9
  )
10
 
 
11
  st.title("Facial Keypoints Detection")
12
  st.write("Upload a face image and the model will predict facial keypoints.")
13
 
 
2
  import numpy as np
3
  import tensorflow as tf
4
  from PIL import Image
5
+ import os
6
+
7
+ BASE_DIR = os.path.dirname(__file__)
8
+ MODEL_PATH = os.path.join(BASE_DIR, "facial_keypoints_resnet.h5")
9
 
10
  st.set_page_config(
11
  page_title="Facial Keypoints Detection",
12
  layout="centered"
13
  )
14
 
15
+
16
  st.title("Facial Keypoints Detection")
17
  st.write("Upload a face image and the model will predict facial keypoints.")
18