Update app.py

app.py

@@ -13,6 +13,12 @@ feature_extractor = SegformerFeatureExtractor.from_pretrained(
 model = SegformerForSemanticSegmentation.from_pretrained(
     "segments-tobias/segformer-b0-finetuned-segments-sidewalk")
 
+url = "https://segmentsai-prod.s3.eu-west-2.amazonaws.com/assets/admin-tobias/439f6843-80c5-47ce-9b17-0b2a1d54dbeb.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
+inputs = feature_extractor(images=image, return_tensors="pt")
+outputs = model(**inputs)
+logits = outputs.logits 
+
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [