Spaces:
Sleeping
Sleeping
jioji0
commited on
Commit
•
8a693b4
1
Parent(s):
ed26df5
img add
Browse files
app.py
CHANGED
|
@@ -11,8 +11,7 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
|
|
| 11 |
"nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
|
| 12 |
)
|
| 13 |
model = TFSegformerForSemanticSegmentation.from_pretrained(
|
| 14 |
-
"nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
|
| 15 |
-
from_pt=True
|
| 16 |
)
|
| 17 |
|
| 18 |
def ade_palette():
|
|
@@ -106,7 +105,7 @@ demo = gr.Interface(fn=sepia,
|
|
| 106 |
inputs=gr.Image(shape=(400, 600)),
|
| 107 |
#inputs=gr.Image(type='pil', prepeocessing_function=prepro_img),
|
| 108 |
outputs=['plot'],
|
| 109 |
-
examples=["city-1c.jpg","city-2","city-3"],
|
| 110 |
allow_flagging='never')
|
| 111 |
|
| 112 |
|
|
|
|
| 11 |
"nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
|
| 12 |
)
|
| 13 |
model = TFSegformerForSemanticSegmentation.from_pretrained(
|
| 14 |
+
"nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
|
|
|
|
| 15 |
)
|
| 16 |
|
| 17 |
def ade_palette():
|
|
|
|
| 105 |
inputs=gr.Image(shape=(400, 600)),
|
| 106 |
#inputs=gr.Image(type='pil', prepeocessing_function=prepro_img),
|
| 107 |
outputs=['plot'],
|
| 108 |
+
examples=["city-1c.jpg","city-2.jpg","city-3.jpg"],
|
| 109 |
allow_flagging='never')
|
| 110 |
|
| 111 |
|