Update app.py

app.py

@@ -203,18 +203,21 @@ def blend_vsfx(video_in, audio_result):
 def infer(video_in, chosen_model):
     image_in = extract_firstframe(video_in)
     caption = get_caption(image_in)
-    if chosen_model == "MAGNet" :
-        audio_result = get_magnet(caption)
-    elif chosen_model == "AudioLDM-2" : 
-        audio_result = get_audioldm(caption)
-    elif chosen_model == "AudioGen" :
-        audio_result = get_audiogen(caption)
-    elif chosen_model == "Tango" :
-        audio_result = get_tango(caption)
-    elif chosen_model == "Tango 2" :
-        audio_result = get_tango2(caption)
-    elif chosen_model == "Stable Audio Open" :
-        audio_result = get_stable_audio_open(caption)
+    try:
+        if chosen_model == "MAGNet" :
+            audio_result = get_magnet(caption)
+        elif chosen_model == "AudioLDM-2" : 
+            audio_result = get_audioldm(caption)
+        elif chosen_model == "AudioGen" :
+            audio_result = get_audiogen(caption)
+        elif chosen_model == "Tango" :
+            audio_result = get_tango(caption)
+        elif chosen_model == "Tango 2" :
+            audio_result = get_tango2(caption)
+        elif chosen_model == "Stable Audio Open" :
+            audio_result = get_stable_audio_open(caption)
+    except:
+        raise gr.Error(f"an error occured with {chosen_model}")
     
     final_res = blend_vsfx(video_in, audio_result)
     return gr.update(value=caption, interactive=True), gr.update(interactive=True), audio_result, final_res
@@ -222,19 +225,21 @@ def infer(video_in, chosen_model):
 def retry(edited_prompt, video_in, chosen_model):
     image_in = extract_firstframe(video_in)
     caption = edited_prompt
-    if chosen_model == "MAGNet" :
-        audio_result = get_magnet(caption)
-    elif chosen_model == "AudioLDM-2" : 
-        audio_result = get_audioldm(caption)
-    elif chosen_model == "AudioGen" :
-        audio_result = get_audiogen(caption)
-    elif chosen_model == "Tango" :
-        audio_result = get_tango(caption)
-    elif chosen_model == "Tango 2" :
-        audio_result = get_tango2(caption)
-    elif chosen_model == "Stable Audio Open" :
-        audio_result = get_stable_audio_open(caption)
-    
+    try:
+        if chosen_model == "MAGNet" :
+            audio_result = get_magnet(caption)
+        elif chosen_model == "AudioLDM-2" : 
+            audio_result = get_audioldm(caption)
+        elif chosen_model == "AudioGen" :
+            audio_result = get_audiogen(caption)
+        elif chosen_model == "Tango" :
+            audio_result = get_tango(caption)
+        elif chosen_model == "Tango 2" :
+            audio_result = get_tango2(caption)
+        elif chosen_model == "Stable Audio Open" :
+            audio_result = get_stable_audio_open(caption)
+    except:
+        raise gr.Error(f"an error occured with {chosen_model}")
     final_res = blend_vsfx(video_in, audio_result)
     return audio_result, final_res