Update app.py

app.py

@@ -8,15 +8,18 @@ from kornia_moons.feature import *
 import gradio as gr
 
 def load_torch_image(fname):
-    img = K.image_to_tensor(cv2.imread(fname), False).float() /255.
+    img = K.image_to_tensor(fname, False).float() /255.
     img = K.color.bgr_to_rgb(img)
     return img
 def inference(file1,file2):
     fname1 = file1.name
     fname2 = file2.name
-
-    img1 = load_torch_image(fname1)
-    img2 = load_torch_image(fname2)
+    im1 = cv2.imread(fname1)                 
+    im1re = cv2.resize(im1, (512, 512))  
+    im2 = cv2.imread(fname2)                 
+    im2re = cv2.resize(im2, (512, 512))       
+    img1 = load_torch_image(im1re)
+    img2 = load_torch_image(im2re)
 
 
     matcher = KF.LoFTR(pretrained='outdoor')