Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -149,10 +149,8 @@ def inference_scribble(image):
|
|
| 149 |
return [Image.fromarray((mask_colors * 0.6 + test_image * 0.4).astype('uint8'), 'RGB'),
|
| 150 |
Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
|
| 151 |
|
| 152 |
-
demo = gr.Interface(
|
| 153 |
-
|
| 154 |
-
inputs=
|
| 155 |
-
gr.ImageMask(label="[Stroke] Draw on Image", type='pil'),
|
| 156 |
outputs=[
|
| 157 |
gr.outputs.Image(type="pil", label="Mask with Image"),
|
| 158 |
gr.outputs.Image(type="pil", label="Mask")
|
|
|
|
| 149 |
return [Image.fromarray((mask_colors * 0.6 + test_image * 0.4).astype('uint8'), 'RGB'),
|
| 150 |
Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
|
| 151 |
|
| 152 |
+
demo = gr.Interface(fn=inference_scribble,
|
| 153 |
+
inputs=gr.ImageMask(label="[Stroke] Draw on Image", type='pil'),
|
|
|
|
|
|
|
| 154 |
outputs=[
|
| 155 |
gr.outputs.Image(type="pil", label="Mask with Image"),
|
| 156 |
gr.outputs.Image(type="pil", label="Mask")
|