Update app.py

app.py

@@ -7,11 +7,9 @@ from PIL import Image
 import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
-# 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")
+feature_extractor = SegformerFeatureExtractorExtractor.from_pretrained("prem-timsina/segformer-b0-finetuned-food")
+model = TFSegformerForSemanticSegmentation.from_pretrained("prem-timsina/segformer-b0-finetuned-food")
 
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""