Amould commited on
Commit
dd61989
·
verified ·
1 Parent(s): fe387c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -80,17 +80,16 @@ with gr.Blocks() as demo:
80
  elem_id = "image-in",
81
  )
82
 
83
- '''
84
  model_list = gr.Dropdown(
85
- ["Imagenet-wild", "Imagenet-affine"], label="Model", info="select a model"
86
  )
87
- '''
88
  out_image = gr.Image(label = "ٌRegistered image",
89
  #source = "upload",
90
  #type = "filepath",
91
  elem_id = "image-out"
92
  )
93
- inputs = [image_1, image_2]#, model_list]
94
  iface = gr.Interface(fn=imgnt_reg, inputs=inputs,outputs=out_image,
95
  title="Imagenet registration V2",
96
  description="Upload 2 images to generate a registered one:",
 
80
  elem_id = "image-in",
81
  )
82
 
 
83
  model_list = gr.Dropdown(
84
+ ["Additive_Recurence", "Rawblock"], label="Model", info="select a model"
85
  )
86
+
87
  out_image = gr.Image(label = "ٌRegistered image",
88
  #source = "upload",
89
  #type = "filepath",
90
  elem_id = "image-out"
91
  )
92
+ inputs = [image_1, image_2, model_list]
93
  iface = gr.Interface(fn=imgnt_reg, inputs=inputs,outputs=out_image,
94
  title="Imagenet registration V2",
95
  description="Upload 2 images to generate a registered one:",