Update app.py

app.py

@@ -22,6 +22,15 @@ def setup_model(args):
 
 def main(img):
 
+    # Checking Image Aspect Ratio
+    with Image.open(img) as im:
+        #image size
+        ratio_width = im.size[0]
+        ratio_height = im.size[1]
+    print(f"Aspect ratio: {ratio_width}x{ratio_height} px")
+    if ratio_width >= 1026 or ratio_height >= 1026:
+        raise gr.Error("Image aspect ratio must not exceed width: 1024 px and height: 1024 px.")
+
     # setup folder and path
     #basewidth = 256
     #wpercent = (basewidth/float(img.size[0]))
@@ -336,13 +345,15 @@ if __name__ == '__main__':
     article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2209.11345' target='_blank'>Swin2SR: SwinV2 Transformer for Compressed Image Super-Resolution and Restoration</a> | <a href='https://github.com/mv-lab/swin2sr' target='_blank'>Github Repo</a></p>"
 
     examples= glob.glob("testsets/real-inputs/*.jpg")
-    gr.Interface(
-        main, 
-        #gr.Image().style(full_width=True, height=60),
-        gr.inputs.Image(type="pil", label="Input").style(height=260),
-        gr.inputs.Image(type="pil", label="Ouput").style(height=240),
-        title=title,
-        description=description,
-        article=article,
-        examples=examples,
-        ).launch(enable_queue=True)
+    demo = gr.Interface(
+            main, 
+            #gr.Image().style(full_width=True, height=60),
+            gr.Image(type="pil", label="Input").style(height=260),
+            gr.Image(type="pil", label="Ouput").style(height=240),
+            title=title,
+            description=description,
+            article=article,
+            examples=examples,
+            ).launch(enable_queue=True)
+    demo.queue()
+    demo.launch()