Update app.py

app.py

@@ -62,7 +62,7 @@ for filename, url in sample_videos_urls.items():
         print(f"{filename} already exists. Skipping download.")
 
 csv.field_size_limit(100000000)
-options = ['language', "sound_and_language", "sound"] # Changed second to initial
+options = ['language', "sound-language", "sound"] # Changed second to initial
 load_size = 224
 plot_size = 224
 
@@ -74,7 +74,12 @@ video_output2 = gr.Video(label="Multi-Head Audio Video Attention (Only Availible
                          height=480)
 video_output3 = gr.Video(label="Visual Features", height=480)
 
-models = {o: LitAVAligner.from_pretrained(f"mhamilton723/DenseAV-{o}") for o in options}
+# models = {o: LitAVAligner.from_pretrained(f"mhamilton723/DenseAV-{o}") for o in options}
+models = {o: None for o in options}
+
+for o in options:
+    print(o)
+    models[o] = LitAVAligner.from_pretrained(f"mhamilton723/DenseAV-{o}")
 
 def process_video(video, model_option):
     # model = models[model_option].cuda()