Update app.py

app.py

@@ -261,7 +261,7 @@ def generate_video(prompt, seed, image=None):
                 parameters_level=True,
                 compiler_transformer=False,
             ),
-        ).to("cuda")
+        )
         _predictor.initialize()
         logger.info("Predictor initialized")
     with torch.no_grad():