Remove direct cuda reference

app.py

@@ -36,7 +36,7 @@ def load_model():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = torch.load(ckpt_path, map_location=device)
     model = disable_dropout(model)
-    model.to("cuda")
+    model.to(device)
     model.eval()
     return model