Spaces:

5Grains
/

Week_10_Second

Sleeping

5Grains commited on Nov 12, 2023

Commit

8404861

1 Parent(s): 627f996

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmen
 # 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():

 # Load model directly
 from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
+feature_extractor = AutoFeatureExtractor.from_pretrained("prem-timsina/segformer-b0-finetuned-food")
 model = SegformerForSemanticSegmentation.from_pretrained("prem-timsina/segformer-b0-finetuned-food")
 def ade_palette():