fix typo, try gr.ImageMask

app.py

@@ -71,8 +71,9 @@ def query_image(img):
 # demo
 demo = gr.Interface(
     query_image,
-    inputs=[gr.Image(type="pil").style(full_width=True, height=256)],
-    outputs=[gr.image().style(full_width=True, height=256)],
+    #inputs=[gr.Image(type="pil").style(full_width=True, height=256)],
+    inputs=[gr.ImageMask(type="pil").style(full_width=True, height=256)],
+    outputs=[gr.Image().style(full_width=True, height=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)",
     examples=[["073.png"], ["356.png"], ["599.png"], ["630.png"], ["673.png"],