Update app.py

app.py

@@ -1,15 +1,15 @@
 import gradio as gr
-from transformers import pipeline, AutoFeatureExtractor, AutoModelForImageClassification
+from transformers import pipeline, AutoFeatureExtractor, SegformerForSemanticSegmentation
 
-extractor = AutoFeatureExtractor.from_pretrained("nateraw/vit-age-classifier")
+extractor = AutoFeatureExtractor.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
 
-model = AutoModelForImageClassification.from_pretrained("nateraw/vit-age-classifier")
+model = SegformerForSemanticSegmentation.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
 
-pipe = pipeline(task="image-classification", 
+pipe = pipeline(task="image-segmentation", 
                 model=model,
-                )
+                feature_extractor=extractor)
 
-gr.Interface.load(pipe, 
-                    title="age class",
+gr.Interface.from_pipeline(pipe, 
+                    title="Segformer",
                     description="Add description",
                     ).launch()