EUNSEO56 commited on
Commit
6ccd3cf
·
1 Parent(s): b9da158

Update app.py

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