use kornia io to load images

app.py

@@ -5,9 +5,12 @@ from kornia.contrib import ImageStitcher
 import kornia.feature as KF
 import torch
 
-def inference(file_1, file_2):   
-    img_1 = K.image_to_tensor(io.imread(file_1), False).float() / 255.
-    img_2 = K.image_to_tensor(io.imread(file_2), False).float() / 255.
+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_1[None]  # 1xCxHxW / fp32 / [0, 1]
+    
     IS = ImageStitcher(KF.LoFTR(pretrained='outdoor'), estimator='ransac')
     with torch.no_grad():
         result = IS(img_1, img_2)