Update app.py

app.py

@@ -58,7 +58,13 @@ def generate_video(prompt: str):
     ensure_model_downloaded()
 
     # Load model
-    manager = ModelManager(pretrained_model_dir=WAN_MODEL_PATH)
+    manager = ModelManager(
+      file_path_list=[WAN_MODEL_PATH],
+      torch_dtype=torch.float16,
+      device="cuda"
+     )
+
+   
     model = manager.load_model()
 
     # Set up pipeline