emirhanno commited on
Commit
470c68c
1 Parent(s): afff12f
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -47,11 +47,11 @@ def segment_hair(image):
47
  # Gradio interface
48
  iface = gr.Interface(
49
  fn=segment_hair,
50
- inputs=gr.inputs.Image(type="numpy"),
51
- outputs=gr.outputs.Image(type="numpy"),
52
  title="Hair Segmentation",
53
  description="Upload an image to segment the hair and generate a mask.",
54
- examples=["data/example.jpeg"]
55
  )
56
 
57
  if __name__ == "__main__":
 
47
  # Gradio interface
48
  iface = gr.Interface(
49
  fn=segment_hair,
50
+ inputs=gr.Image(type="numpy"),
51
+ outputs=gr.Image(type="numpy"),
52
  title="Hair Segmentation",
53
  description="Upload an image to segment the hair and generate a mask.",
54
+ examples=["example.jpg"]
55
  )
56
 
57
  if __name__ == "__main__":