Spaces:
Runtime error
Runtime error
Update app.py (#2)
Browse files- Update app.py (8122051678094a8d979ed15f791ba98ffbe5d28d)
- Upload ADE_val_00000001.jpeg (70270c6bd74d73a80cffb771d18e8720c04da340)
Co-authored-by: Sayak Paul <[email protected]>
- ADE_val_00000001.jpeg +0 -0
- app.py +2 -2
ADE_val_00000001.jpeg
ADDED
![]() |
app.py
CHANGED
@@ -9,7 +9,7 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
|
|
9 |
"nvidia/segformer-b5-finetuned-ade-640-640"
|
10 |
)
|
11 |
model = TFSegformerForSemanticSegmentation.from_pretrained(
|
12 |
-
"nvidia/segformer-b5-finetuned-ade-640-640"
|
13 |
)
|
14 |
|
15 |
def ade_palette():
|
@@ -196,6 +196,6 @@ def sepia(input_img):
|
|
196 |
pred_img = pred_img.astype(np.uint8)
|
197 |
return pred_img
|
198 |
|
199 |
-
demo = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image")
|
200 |
|
201 |
demo.launch()
|
|
|
9 |
"nvidia/segformer-b5-finetuned-ade-640-640"
|
10 |
)
|
11 |
model = TFSegformerForSemanticSegmentation.from_pretrained(
|
12 |
+
"nvidia/segformer-b5-finetuned-ade-640-640"
|
13 |
)
|
14 |
|
15 |
def ade_palette():
|
|
|
196 |
pred_img = pred_img.astype(np.uint8)
|
197 |
return pred_img
|
198 |
|
199 |
+
demo = gr.Interface(sepia, gr.Image(shape=(200, 200)), "image", examples=["ADE_val_00000001.jpeg"])
|
200 |
|
201 |
demo.launch()
|