resize the output image

app.py

@@ -57,7 +57,8 @@ def query_image(img):
 demo = gr.Interface(
     query_image,
     inputs=[gr.Image(type="pil")],
-    outputs="image",
+    #outputs="image",
+    outputs=[gr.Image().style(full_width=True, height=128, width=128)],
     title="Skyguard: segmentador de glaciares de roca  🛰️ +️ 🛡️ ️",
     description="Modelo de segmentación de imágenes para detectar glaciares de roca.<br> Se entrenó un modelo [nvidia/SegFormer](https://huggingface.co/nvidia/mit-b0) con _fine-tuning_ en el [rock-glacier-dataset](https://huggingface.co/datasets/alkzar90/rock-glacier-dataset)"
 )