Update app.py

app.py

@@ -12,11 +12,16 @@ from hfserver import HuggingFaceDatasetSaver, HuggingFaceDatasetJSONSaver
 # download data from huggingface dataset
 # dataset = load_dataset("quantumiracle-git/robotinder-data")
 
-# download data from google drive
-url = 'https://drive.google.com/drive/folders/1JuNQS4R7axTezWj1x4KRAuRt_L26ApxA?usp=sharing'
-output = './'
-id = url.split('/')[-1]
-os.system(f"gdown --id {id} -O {output} --folder --no-cookies --remaining-ok")
+LOAD_DATA_GOOGLE_DRIVE = True
+if LOAD_DATA_GOOGLE_DRIVE:  # download data from google drive
+    url = 'https://drive.google.com/drive/folders/1JuNQS4R7axTezWj1x4KRAuRt_L26ApxA?usp=sharing'  # './processed/' folder in google drive
+    output = './'
+    id = url.split('/')[-1]
+    os.system(f"gdown --id {id} -O {output} --folder --no-cookies --remaining-ok")
+    VIDEO_PATH = 'processed'
+else:  # local data
+    VIDEO_PATH = 'robotinder-data'
+
 
 def video_identity(video):
     return video
@@ -25,7 +30,6 @@ def nan():
     return None
 
 FORMAT = ['mp4', 'gif'][1]
-VIDEO_PATH = 'robotinder-data'
 
 def get_huggingface_dataset():
     try: