Spaces:

5Grains
/

Week_10_Second

Sleeping

5Grains commited on Nov 12, 2023

Commit

b53ea11

1 Parent(s): 73e5dcd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,9 +8,8 @@ import tensorflow as tf
 # Load model directly
 from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
-extractor = AutoFeatureExtractor.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512")
-model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512")
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [

 # Load model directly
 from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
+extractor = AutoFeatureExtractor.from_pretrained("segments-tobias/segformer-b0-finetuned-segments-sidewalk")
+model = SegformerForSemanticSegmentation.from_pretrained("segments-tobias/segformer-b0-finetuned-segments-sidewalk")
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [