matplotlib clf, image input reshaping to None

app.py

@@ -23,16 +23,18 @@ def pump_matching(img1, img2, trained_with_st=False, scale=300, max_scale=1, max
     corres = Main().run_from_args_with_images(img1, img2, args)
         
     fig1 = plt.figure(1)
-    plt.imshow(img1)
+    plt.clf()
     ax1 = plt.gca()
+    ax1.imshow(img1)
     ax1.axis('off')
-    plt.tight_layout()  
+    plt.tight_layout(pad=0)  
     
     fig2 = plt.figure(2)
-    plt.imshow(img2)
+    plt.clf()
     ax2 = plt.gca()
+    ax2.imshow(img2)
     ax2.axis('off')
-    plt.tight_layout()  
+    plt.tight_layout(pad=0) 
        
     from tools.viz import plot_grid
     if corres.shape[-1] > 4:
@@ -54,8 +56,8 @@ article = "<p style='text-align: center'><a href='https://github.com/naver/pump'
 iface = gr.Interface(
     fn=pump_matching,
     inputs=[
-        gr.inputs.Image(shape=(1024, 1024), type="pil", label="First Image"),
-        gr.inputs.Image(shape=(1024, 1024), type="pil", label="Second Image"),
+        gr.inputs.Image(shape=None, type="pil", label="First Image"),
+        gr.inputs.Image(shape=None, type="pil", label="Second Image"),
         gr.inputs.Checkbox(default=False, label="Use the model trained with style transfer"),
         #gr.inputs.Slider(minimum=300, maximum=600, default=400, step=1, label="Original test scale"),
         #gr.inputs.Slider(minimum=1, maximum=4, default=1, step=0.1, label="Multi Scale Testing - maximum scale (makes it slower)"),