segformer extracter

app.py

@@ -1,4 +1,4 @@
-from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
+from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 from torch import nn
 import numpy as np
 import matplotlib.pyplot as plt
@@ -7,7 +7,7 @@ from PIL import Image
 import gradio as gr
 
 def seg(image):
-    feature_extractor = AutoFeatureExtractor.from_pretrained("nvidia/mit-b5")
+    feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/mit-b5")
     model = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b5")
     print(model)