Update app.py

app.py

@@ -216,11 +216,12 @@ def infer(video_in, chosen_model):
             audio_result = get_tango2(caption)
         elif chosen_model == "Stable Audio Open" :
             audio_result = get_stable_audio_open(caption)
+        final_res = blend_vsfx(video_in, audio_result)
+        return gr.update(value=caption, interactive=True), gr.update(interactive=True), audio_result, final_res
     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
+    
 
 def retry(edited_prompt, video_in, chosen_model):
     image_in = extract_firstframe(video_in)
@@ -238,10 +239,11 @@ def retry(edited_prompt, video_in, chosen_model):
             audio_result = get_tango2(caption)
         elif chosen_model == "Stable Audio Open" :
             audio_result = get_stable_audio_open(caption)
+        final_res = blend_vsfx(video_in, audio_result)
+        return audio_result, final_res
     except:
         raise gr.Error(f"an error occured with {chosen_model}")
-    final_res = blend_vsfx(video_in, audio_result)
-    return audio_result, final_res
+    
 
 def refresh():
     return gr.update(value=None, interactive=False), gr.update(interactive=False), gr.update(value=None), gr.update(value=None)