Update app.py

app.py

@@ -91,19 +91,19 @@ def generate_video(prompt: str):
     ensure_model_downloaded()
 
     # Load model using patched manager
-    # manager = PatchedModelManager(
-    #     file_path_list=[WAN_MODEL_PATH],
-    #     torch_dtype=torch.float16,
-    #     device="cuda"
-    # )
+    manager = ModelManager(
+        file_path_list=[WAN_MODEL_PATH],
+        torch_dtype=torch.float16,
+        device="cuda"
+    )
 
     
 
-    manager = PatchedModelManager(
-       file_path_list=[WAN_MODEL_PATH],
-       torch_dtype=torch.float16,
-       device="cuda" 
-     )
+    # manager = PatchedModelManager(
+    #    file_path_list=[WAN_MODEL_PATH],
+    #    torch_dtype=torch.float16,
+    #    device="cuda" 
+    #  )
    
     model =  manager.load_model(WAN_MODEL_PATH)