Update app.py

app.py

@@ -5,11 +5,15 @@ import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
 import tensorflow as tf
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+pipe = pipeline("image-segmentation", model="nvidia/segformer-b0-finetuned-cityscapes-512-1024")
 # Load model directly
 from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
 
-extractor = AutoFeatureExtractor.from_pretrained("segments-tobias/segformer-b0-finetuned-segments-sidewalk")
-model = SegformerForSemanticSegmentation.from_pretrained("segments-tobias/segformer-b0-finetuned-segments-sidewalk")
+extractor = AutoFeatureExtractor.from_pretrained("nvidia/segformer-b0-finetuned-cityscapes-512-1024")
+model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b0-finetuned-cityscapes-512-1024")
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [