Update app.py

app.py

@@ -6,11 +6,12 @@ import kornia.feature as KF
 import torch
 import numpy as np
 
-def inference(file_1, file_2):
-    img_1: Tensor = K.io.load_image(file_1, K.io.ImageLoadType.RGB32)
-    img_1 = img_1[None]  # 1xCxHxW / fp32 / [0, 1]
-    img_2: Tensor = K.io.load_image(file_2, K.io.ImageLoadType.RGB32)
-    img_2 = img_2[None]  # 1xCxHxW / fp32 / [0, 1]
+def inference(img_1, img_2):
+    # Convert numpy arrays to Tensors
+    img_1: Tensor = K.image_to_tensor(img_1, keepdim=False).float() / 255.0
+    img_1 = img_1.unsqueeze(0)  # Add batch dimension
+    img_2: Tensor = K.image_to_tensor(img_2, keepdim=False).float() / 255.0
+    img_2 = img_2.unsqueeze(0)  # Add batch dimension
     
     IS = ImageStitcher(KF.LoFTR(pretrained='outdoor'), estimator='ransac')
     with torch.no_grad():