Resize input - output display 256x256

app.py

@@ -56,9 +56,8 @@ def query_image(img):
 
 demo = gr.Interface(
     query_image,
-    inputs=[gr.Image(type="pil")],
-    #outputs="image",
-    outputs=[gr.Image().style(full_width=True, height=128, width=128)],
+    inputs=[gr.Image(type="pil").style(full_width=True, height=256, width=256)],
+    outputs=[gr.Image().style(full_width=True, height=256, width=256)],
     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)"
 )