andresgtn commited on
Commit
fed4914
·
1 Parent(s): 7392a8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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
- interface = gr.Interface(classify, gr.Image(), 'image', examples=sample_images)# FILL HERE
 
 
 
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)