alkzar90 commited on
Commit
334cac1
1 Parent(s): a5218e8

center input-outpuut block

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -71,8 +71,7 @@ def query_image(img):
71
  # demo
72
  demo = gr.Interface(
73
  query_image,
74
- #inputs=[gr.Image(type="pil").style(full_width=True, height=256)],
75
- inputs=[gr.ImageMask(type="pil").style(full_width=True, height=256)],
76
  outputs=[gr.Image().style(full_width=True, height=256)],
77
  title="Skyguard: segmentador de glaciares de roca 🛰️ +️ 🛡️ ️",
78
  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)",
71
  # demo
72
  demo = gr.Interface(
73
  query_image,
74
+ inputs=[gr.Image(type="pil").style(full_width=True, height=256)],
 
75
  outputs=[gr.Image().style(full_width=True, height=256)],
76
  title="Skyguard: segmentador de glaciares de roca 🛰️ +️ 🛡️ ️",
77
  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)",