Update app.py

app.py

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