fixing snapshot model download

app.py

@@ -215,7 +215,7 @@ with gr.Blocks(title='Shark Patrol',theme=gr.themes.Soft(),live=True,) as demo:
             paths = sorted(pathlib.Path('videos_example/').rglob('*.mp4'))
             example_images = gr.Dataset(components=[video_input],
                                     samples=[[path.as_posix()]
-                                             for path in paths if 'videos_side_by_side' not in str(path)])
+                                             for path in paths if 'raw_videos' not in str(path)])
 
     video_button.click(analyze_video_parallel, inputs=video_input, outputs=video_output)
 

inference.py

@@ -80,7 +80,7 @@ classes= ['Beach',
 REPO_ID = "SharkSpace/maskformer_model"
 FILENAME = "mask2former"
 
-# snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')
+snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')