made a mistake earlier now made for semantic segmentation

app.py

@@ -1,14 +1,12 @@
-from transformers import SegformerFeatureExtractor, SegformerForImageClassification
+from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
 from PIL import Image
 import requests
 
 import gradio as gr
 
-
-
 def seg(image):
-    feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/mit-b0")
-    model = SegformerForImageClassification.from_pretrained("nvidia/mit-b0")
+    feature_extractor = AutoFeatureExtractor.from_pretrained("nvidia/mit-b0")
+    model = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b0")
     print(model)
     
     inputs = feature_extractor(images=image, return_tensors="pt")