Update app.py

app.py

@@ -62,7 +62,7 @@ def load_custom_model(model_key):
     model_info = MODEL_OPTIONS[model_key]
     # Pass model_name to config for correct model instantiation
     config = {"model": {"name": model_info["model_name"]}}
-    model = XrayReg.load_from_checkpoint(model_info["ckpt"])
+    model = XrayReg.load_from_checkpoint(model_info["ckpt"], map_location="cpu")
     model = model.model
     model.eval()
     for param in model.parameters():