lorocksUMD commited on
Commit
083f600
·
verified ·
1 Parent(s): 9cda79e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -62,7 +62,7 @@ for filename, url in sample_videos_urls.items():
62
  print(f"{filename} already exists. Skipping download.")
63
 
64
  csv.field_size_limit(100000000)
65
- options = ['language', "sound_and_language", "sound"]
66
  load_size = 224
67
  plot_size = 224
68
 
@@ -131,7 +131,7 @@ def process_video(video, model_option):
131
  sample_rate,
132
  temp_video_path_1)
133
 
134
- if model_option == "sound-language":
135
  temp_video_path_2 = tempfile.mktemp(suffix='.mp4')
136
 
137
  plot_2head_attention_video(
 
62
  print(f"{filename} already exists. Skipping download.")
63
 
64
  csv.field_size_limit(100000000)
65
+ options = ['language', "sound_and_language", "sound"] # Changed second to initial
66
  load_size = 224
67
  plot_size = 224
68
 
 
131
  sample_rate,
132
  temp_video_path_1)
133
 
134
+ if model_option == "sound_and_language": # change to _ _
135
  temp_video_path_2 = tempfile.mktemp(suffix='.mp4')
136
 
137
  plot_2head_attention_video(