Update app.py

app.py

@@ -14,6 +14,18 @@ except Exception as e:
     print(f"GPU warmup failed: {e}")
 os.environ["GRADIO_TEMP_DIR"] = "./tmp"
 
+try:
+        global pipe, fantasytalking, wav2vec_processor, wav2vec  
+        
+        pipe,fantasytalking,wav2vec_processor,wav2vec = load_models(args)
+        output_path=main(
+            args,pipe,fantasytalking,wav2vec_processor,wav2vec
+        )
+        return output_path  # Ensure the output path is returned
+    except Exception as e:
+        print(f"Error during processing: {str(e)}")
+        raise gr.Error(f"Error during processing: {str(e)}")
+
 pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
 @spaces.GPU(duration=1200)
 def generate_video(
@@ -53,17 +65,7 @@ def generate_video(
         seed=seed,
     )
 
-    try:
-        global pipe, fantasytalking, wav2vec_processor, wav2vec  
-        
-        pipe,fantasytalking,wav2vec_processor,wav2vec = load_models(args)
-        output_path=main(
-            args,pipe,fantasytalking,wav2vec_processor,wav2vec
-        )
-        return output_path  # Ensure the output path is returned
-    except Exception as e:
-        print(f"Error during processing: {str(e)}")
-        raise gr.Error(f"Error during processing: {str(e)}")
+    
 
 
 def create_args(