Hugging Face's logo

Spaces:
0jung
/
Paper3_Segmentation
Runtime error

App Files Community
0jung commited on
Commit
f274621
•
1 Parent(s): f4e7121

modify

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,10 +9,10 @@ from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmen
9
 
10
 
11
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
12
- "facebook/maskformer-swin-large-ade"
13
  )
14
  model = TFSegformerForSemanticSegmentation.from_pretrained(
15
- "facebook/maskformer-swin-large-ade, , from_pt=True"
16
  )
17
 
18
  def ade_palette():
 
9
 
10
 
11
  feature_extractor = SegformerFeatureExtractor.from_pretrained(
12
+ "nvidia/segformer-b0-finetuned-cityscapes-1024-1024"
13
  )
14
  model = TFSegformerForSemanticSegmentation.from_pretrained(
15
+ "nvidia/segformer-b0-finetuned-cityscapes-1024-1024"
16
  )
17
 
18
  def ade_palette():