Update app.py

app.py

@@ -134,7 +134,7 @@ def generate_t2v_video(self, prompt, lora_alpha, num_inference_steps,
             
 @spaces.GPU(duration=200)
 def generate_video(prompt: str):
-    ensure_model_downloaded()
+   
 
     # Load model using patched manager
 
@@ -142,13 +142,13 @@ def generate_video(prompt: str):
     model_manager = ModelManager(device="cuda")
     base_dir = "model_zoo/PusaV1/Wan2.1-T2V-14B"
             
-    model_files = sorted([os.path.join(self.base_dir, f) for f in os.listdir(self.base_dir) if f.endswith('.safetensors')])
+    model_files = sorted([os.path.join(base_dir, f) for f in os.listdir(base_dir) if f.endswith('.safetensors')])
             
     model_manager.load_models(
                 [
                     model_files,
-                    os.path.join(self.base_dir, "models_t5_umt5-xxl-enc-bf16.pth"),
-                    os.path.join(self.base_dir, "Wan2.1_VAE.pth"),
+                    os.path.join(base_dir, "models_t5_umt5-xxl-enc-bf16.pth"),
+                    os.path.join(base_dir, "Wan2.1_VAE.pth"),
                 ],
                 torch_dtype=torch.bfloat16,
             )
@@ -202,4 +202,8 @@ with gr.Blocks() as demo:
 
     generate_btn.click(fn=generate_video, inputs=prompt_input, outputs=video_output)
 
-demo.launch()
+
+
+if __name__ == "__main__":
+    ensure_model_downloaded()
+    demo.launch(share=True, show_error=True)