forcing torch.load to CPU

app.py

@@ -9,6 +9,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 _old_load = torch.load
 
 def safe_torch_load(*args, **kwargs):
+    args = list(args)
     if len(args) >= 2:
         args[1] = device
     else: