0jung commited on
Commit
f4e7121
1 Parent(s): 43d4af4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
12
  "facebook/maskformer-swin-large-ade"
13
  )
14
  model = TFSegformerForSemanticSegmentation.from_pretrained(
15
- "facebook/maskformer-swin-large-ade"
16
  )
17
 
18
  def ade_palette():
 
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():