Update app.py

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 import os
 
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+# device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # model_id = "hunyuanvideo-community/HunyuanVideo"
 
@@ -33,7 +33,7 @@ def generate(prompt, width=832, height=832, num_inference_steps=30, lora_id=None
     #     pipe,
     #     # residual_diff_threshold=0.2,
     # )
-    # pipe.to("cuda")
+    pipe.to("cuda")
     torch.cuda.empty_cache()
     try:
         output = pipe(