Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ class TargetMaskConvertTransform(ItemTransform):
|
|
49 |
return img, mask
|
50 |
|
51 |
|
52 |
-
repo_id = "joortif/
|
53 |
|
54 |
learn = from_pretrained_fastai(repo_id)
|
55 |
|
@@ -61,6 +61,6 @@ interface = gr.Interface(
|
|
61 |
fn=segment_image,
|
62 |
inputs=gr.Image(),
|
63 |
outputs=gr.Image(),
|
64 |
-
|
65 |
title="Segmentaci贸n Sem谩ntica con FastAI",
|
66 |
).launch(share=False)
|
|
|
49 |
return img, mask
|
50 |
|
51 |
|
52 |
+
repo_id = "joortif/unet-resnet34-segmentation"
|
53 |
|
54 |
learn = from_pretrained_fastai(repo_id)
|
55 |
|
|
|
61 |
fn=segment_image,
|
62 |
inputs=gr.Image(),
|
63 |
outputs=gr.Image(),
|
64 |
+
examples=["color184.jpg","color154.jpg","color180.jpg"],
|
65 |
title="Segmentaci贸n Sem谩ntica con FastAI",
|
66 |
).launch(share=False)
|