Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ import torch
|
|
4 |
from torch import nn
|
5 |
from transformers import SegformerForSemanticSegmentation, SegformerFeatureExtractor
|
6 |
|
|
|
7 |
#extractor = AutoFeatureExtractor.from_pretrained("andresgtn/segformer-b0-finetuned-ade-64-64-finetuned-semantic-sidewalk")
|
8 |
extractor = SegformerFeatureExtractor()
|
9 |
model = SegformerForSemanticSegmentation.from_pretrained("andresgtn/segformer-b0-finetuned-ade-64-64-finetuned-semantic-sidewalk")
|
@@ -35,6 +36,9 @@ sample_images = [["https://s3.amazonaws.com/moonup/production/uploads/1664719956
|
|
35 |
["https://s3.amazonaws.com/moonup/production/uploads/1664719956737-611f9702593efbee33a4f7c9.png"]]
|
36 |
|
37 |
# define gradio interface
|
38 |
-
|
|
|
|
|
|
|
39 |
|
40 |
interface.launch(debug=False)
|
|
|
4 |
from torch import nn
|
5 |
from transformers import SegformerForSemanticSegmentation, SegformerFeatureExtractor
|
6 |
|
7 |
+
|
8 |
#extractor = AutoFeatureExtractor.from_pretrained("andresgtn/segformer-b0-finetuned-ade-64-64-finetuned-semantic-sidewalk")
|
9 |
extractor = SegformerFeatureExtractor()
|
10 |
model = SegformerForSemanticSegmentation.from_pretrained("andresgtn/segformer-b0-finetuned-ade-64-64-finetuned-semantic-sidewalk")
|
|
|
36 |
["https://s3.amazonaws.com/moonup/production/uploads/1664719956737-611f9702593efbee33a4f7c9.png"]]
|
37 |
|
38 |
# define gradio interface
|
39 |
+
title = "Image segmentation for sidewalks"
|
40 |
+
description = "Drop an image of a sidewalk"
|
41 |
+
interface = gr.Interface(classify, gr.Image(), 'image', examples=sample_images,
|
42 |
+
description=description, title=title)# FILL HERE
|
43 |
|
44 |
interface.launch(debug=False)
|