Update app.py

app.py

@@ -7,6 +7,17 @@ import gradio as gr
 import os
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
+import torch
+
+# 🔥 GLOBAL PATCH: force all torch.load to CPU
+_original_torch_load = torch.load
+
+def _cpu_only_torch_load(*args, **kwargs):
+    kwargs.setdefault("map_location", torch.device("cpu"))
+    return _original_torch_load(*args, **kwargs)
+
+torch.load = _cpu_only_torch_load
+