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)
|