Update app.py

app.py

@@ -1,7 +1,15 @@
 import gradio as gr
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
-              
-gr.Interface.load("huggingface/google/pegasus-xsum", 
-                           title="Summarization using pegasus",
-                           description="Summarization description",
-                           ).launch()
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM, AutoFeatureExtractor, SegformerForSemanticSegmentation
+
+
+extractor = AutoFeatureExtractor.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
+
+model = SegformerForSemanticSegmentation.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
+
+pipe = pipeline(task="feature-extraction", 
+                model=model)
+
+gr.Interface.load(pipe, 
+                    title="Segformer",
+                    description="Add description",
+                    ).launch()