Hugging Face's logo

Spaces:
aupfe08
/
SegFormer_ADE_Segmentation
Runtime error

App Files Community
2
aupfe08 commited on
Commit
1ad1bc3
·
verified ·
1 Parent(s): c83056c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
  import tensorflow as tf
8
  from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
9
 
10
- feature_extractor = SegformerFeatureExtractor.from_pretrained(
11
  "nvidia/segformer-b5-finetuned-ade-640-640"
12
  )
13
  model = SegformerForSemanticSegmentation.from_pretrained(
 
7
  import tensorflow as tf
8
  from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
9
 
10
+ feature_extractor = SegformerImageProcessor.from_pretrained(
11
  "nvidia/segformer-b5-finetuned-ade-640-640"
12
  )
13
  model = SegformerForSemanticSegmentation.from_pretrained(