Spaces:
Sleeping
Sleeping
bugfix
Browse files
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.
|
51 |
-
outputs=gr.
|
52 |
title="Hair Segmentation",
|
53 |
description="Upload an image to segment the hair and generate a mask.",
|
54 |
-
examples=["
|
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__":
|