update

app.py

@@ -45,7 +45,7 @@ def find_matches(image_0, image_1, conf_thres=0.2, border_rm=2, topk=10000):
 
     _config = lower_config(config)
     matcher = EDM(config=_config["edm"])
-    state_dict = torch.load("weights/edm_outdoor.ckpt")["state_dict"]
+    state_dict = torch.load("weights/edm_outdoor.ckpt", map_location=torch.device('cpu'))["state_dict"]
     matcher.load_state_dict(state_dict)
     matcher = matcher.eval()
     
@@ -76,9 +76,9 @@ def find_matches(image_0, image_1, conf_thres=0.2, border_rm=2, topk=10000):
     with torch.no_grad():
         matcher(batch)
 
-    mkpts0 = batch['mkpts0_f'].cpu().numpy()
-    mkpts1 = batch['mkpts1_f'].cpu().numpy()
-    mconf = batch['mconf'].cpu().numpy()
+    mkpts0 = batch['mkpts0_f'].numpy()
+    mkpts1 = batch['mkpts1_f'].numpy()
+    mconf = batch['mconf'].numpy()
     
     mkpts0[:, 0] *= w0_scale
     mkpts0[:, 1] *= h0_scale
@@ -137,4 +137,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch()