kadirnar commited on
Commit
be6c4a0
1 Parent(s): 77b1ac6

update app

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. packages.txt +0 -0
app.py CHANGED
@@ -5,7 +5,7 @@ from kornia.contrib import ImageStitcher
5
  import kornia.feature as KF
6
  import torch
7
 
8
- def enhance(file_1, file_2):
9
  img_1 = K.image_to_tensor(io.imread(file_1), False).float() / 255.
10
  img_2 = K.image_to_tensor(io.imread(file_2), False).float() / 255.
11
  IS = ImageStitcher(KF.LoFTR(pretrained='outdoor'), estimator='ransac')
@@ -34,7 +34,7 @@ outputs = [
34
  title = "Image Stitching using Kornia and LoFTR"
35
 
36
  demo_app = gr.Interface(
37
- fn=enhance,
38
  inputs=inputs,
39
  outputs=outputs,
40
  title=title,
 
5
  import kornia.feature as KF
6
  import torch
7
 
8
+ def inference(file_1, file_2):
9
  img_1 = K.image_to_tensor(io.imread(file_1), False).float() / 255.
10
  img_2 = K.image_to_tensor(io.imread(file_2), False).float() / 255.
11
  IS = ImageStitcher(KF.LoFTR(pretrained='outdoor'), estimator='ransac')
 
34
  title = "Image Stitching using Kornia and LoFTR"
35
 
36
  demo_app = gr.Interface(
37
+ fn=inference,
38
  inputs=inputs,
39
  outputs=outputs,
40
  title=title,
packages.txt DELETED
File without changes