federico commited on
Commit
9f8136b
β€’
1 Parent(s): 1af8725

google drive name update

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -104,12 +104,12 @@ demo = gr.Interface(
104
  )
105
 
106
  if __name__=='__main__':
107
- if not os.path.exists("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 = 'data/head_pose_estimation'
111
  gaze_model = tf.keras.models.load_model(gaze_model_path, custom_objects={"tf": tf})
112
- path_to_model = '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)
 
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)