Update app.py

app.py

@@ -98,7 +98,7 @@ description = f"The underlying model is [this](https://huggingface.co/{CKPT}). Y
 iface = gr.Interface(
     infer,
     inputs="image",
-    outputs="image",
+    outputs=gr.Image().style(height=242),
     title=title,
     description=description,
     allow_flagging="never",