Fixed Imports and Added DDescriptions

app.py

@@ -6,8 +6,11 @@ from common.logger import AverageMeter
 from common.logger import Logger
 from data import download
 from model import chmnet
-from matplotlib import pyplot as plt
+from itertools import product
+import matplotlib
+import matplotlib.patches as patches
 from matplotlib.patches import ConnectionPatch
+from matplotlib import pyplot as plt
 from PIL import Image
 import numpy as np
 import os
@@ -139,11 +142,17 @@ def generate_correspondences(sousrce_image, target_image, min_x=1, max_x=100, mi
 
 
 # GRADIO APP
+title = "Correspondence Matching with Convolutional Hough Matching Networks "
+description = "Performs keypoint transform from on a 7x7 gird from the source image to the target image. Use the sliders to adjust the grid."
+article = "<p style='text-align: center'><a href='https://github.com/juhongm999/chm' target='_blank'>Github Repo</a></p>"
+
 iface = gr.Interface(fn=generate_correspondences, 
                      inputs=[gr.inputs.Image(shape=(240, 240), type='pil'), 
                              gr.inputs.Image(shape=(240, 240), type='pil'),
-                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=15, label='MinX'),
-                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=215, label='MaxX'),
-                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=15, label='MinY'),
-                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=215, label='MaxY')], outputs="plot")
+                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=15, label='Min X'),
+                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=215, label='Max X'),
+                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=15, label='Min Y'),
+                             gr.inputs.Slider(minimum=1, maximum=240, step=1, default=215, label='Max Y')], outputs="plot", enable_queue=True, title=title,
+                            description=description,
+                            article=article,)
 iface.launch()