Spaces:

5Grains
/

Week_10_Second

Sleeping

5Grains commited on Nov 12, 2023

Commit

8cf218e

1 Parent(s): 7edebe4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,8 +10,8 @@ from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmen
 # 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."""

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