Spaces:
Runtime error
Runtime error
federico
commited on
Commit
Β·
9f8136b
1
Parent(s):
1af8725
google drive name update
Browse files
app.py
CHANGED
@@ -104,12 +104,12 @@ demo = gr.Interface(
|
|
104 |
)
|
105 |
|
106 |
if __name__=='__main__':
|
107 |
-
if not os.path.exists("
|
108 |
gdown.download_folder("https://drive.google.com/drive/folders/1nQ1Cb_tBEhWxy183t-mIcVH7AhAfa6NO?usp=drive_link",
|
109 |
use_cookies=False)
|
110 |
-
gaze_model_path = '
|
111 |
gaze_model = tf.keras.models.load_model(gaze_model_path, custom_objects={"tf": tf})
|
112 |
-
path_to_model = '
|
113 |
model = tf.saved_model.load(os.path.join(path_to_model, 'saved_model'))
|
114 |
|
115 |
input_shape_od_model = (512, 512)
|
|
|
104 |
)
|
105 |
|
106 |
if __name__=='__main__':
|
107 |
+
if not os.path.exists("LAEO_demo_data"):
|
108 |
gdown.download_folder("https://drive.google.com/drive/folders/1nQ1Cb_tBEhWxy183t-mIcVH7AhAfa6NO?usp=drive_link",
|
109 |
use_cookies=False)
|
110 |
+
gaze_model_path = 'LAEO_demo_data/head_pose_estimation'
|
111 |
gaze_model = tf.keras.models.load_model(gaze_model_path, custom_objects={"tf": tf})
|
112 |
+
path_to_model = 'LAEO_demo_data/keypoint_detector/centernet_hg104_512x512_kpts_coco17_tpu-32'
|
113 |
model = tf.saved_model.load(os.path.join(path_to_model, 'saved_model'))
|
114 |
|
115 |
input_shape_od_model = (512, 512)
|