federico commited on
Commit
6e7ba4c
β€’
1 Parent(s): b5aadac

corrected examples path

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -125,6 +125,15 @@ if __name__=='__main__':
125
  if not os.path.exists("LAEO_demo_data"):
126
  gdown.download_folder("https://drive.google.com/drive/folders/1nQ1Cb_tBEhWxy183t-mIcVH7AhAfa6NO?usp=drive_link",
127
  use_cookies=False)
 
 
 
 
 
 
 
 
 
128
  gaze_model_path = 'LAEO_demo_data/head_pose_estimation'
129
  gaze_model = tf.keras.models.load_model(gaze_model_path, custom_objects={"tf": tf})
130
  path_to_model = 'LAEO_demo_data/keypoint_detector/centernet_hg104_512x512_kpts_coco17_tpu-32'
 
125
  if not os.path.exists("LAEO_demo_data"):
126
  gdown.download_folder("https://drive.google.com/drive/folders/1nQ1Cb_tBEhWxy183t-mIcVH7AhAfa6NO?usp=drive_link",
127
  use_cookies=False)
128
+
129
+ # Get the list of all files and directories
130
+ path = "LAEO_demo_data/examples"
131
+ dir_list = os.listdir(path)
132
+ print("Files and directories in '", path, "' :")
133
+
134
+ # prints all files
135
+ print(dir_list)
136
+
137
  gaze_model_path = 'LAEO_demo_data/head_pose_estimation'
138
  gaze_model = tf.keras.models.load_model(gaze_model_path, custom_objects={"tf": tf})
139
  path_to_model = 'LAEO_demo_data/keypoint_detector/centernet_hg104_512x512_kpts_coco17_tpu-32'