Update app.py

app.py

@@ -42,6 +42,18 @@ import torch
 torch._dynamo.config.suppress_errors = True
 torch._dynamo.config.disable = True
 
+_original_tensor_to = torch.Tensor.to
+def _coerce_cuda_str(self, *args, **kwargs):
+    if args and isinstance(args[0], torch.device) and args[0].type == "cuda":
+        args = ("cuda", *args[1:])
+    if kwargs.get("device") is not None:
+        dev = kwargs["device"]
+        if isinstance(dev, torch.device) and dev.type == "cuda":
+            kwargs = {**kwargs, "device": "cuda"}
+    return _original_tensor_to(self, *args, **kwargs)
+
+torch.Tensor.to = _coerce_cuda_str
+
 import gradio as gr
 import numpy as np
 from huggingface_hub import hf_hub_download, snapshot_download