Hugging Face's logo

Spaces:
danu9327
/
333
Sleeping

App Files Community
danu9327 commited on
Commit
de3b940
•
1 Parent(s): 241c7ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,8 +11,7 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
11
  "nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
12
  )
13
  model = TFSegformerForSemanticSegmentation.from_pretrained(
14
- "nvidia/segformer-b2-finetuned-cityscapes-1024-1024",
15
- from_pt=True
16
  )
17
 
18
  def ade_palette():
 
11
  "nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
12
  )
13
  model = TFSegformerForSemanticSegmentation.from_pretrained(
14
+ "nvidia/segformer-b2-finetuned-cityscapes-1024-1024"
 
15
  )
16
 
17
  def ade_palette():