Hugging Face's logo

Spaces:
GImInI
/
H3
Sleeping

App Files Community
GImInI commited on
Commit
a01cc0f
·
1 Parent(s): 3b435ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
11
  "nickmuchi/segformer-b4-finetuned-segments-sidewalk"
12
  )
13
  model = TFSegformerForSemanticSegmentation.from_pretrained(
14
- "nickmuchi/segformer-b4-finetuned-segments-sidewalk"
15
  )
16
 
17
  def ade_palette():
 
11
  "nickmuchi/segformer-b4-finetuned-segments-sidewalk"
12
  )
13
  model = TFSegformerForSemanticSegmentation.from_pretrained(
14
+ "nickmuchi/segformer-b4-finetuned-segments-sidewalk", from_pt=True
15
  )
16
 
17
  def ade_palette():