1

app.py

@@ -8,9 +8,11 @@ model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b3-fi
 # url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 # image = Image.open(requests.get(url, stream=True).raw)
 
-images = ['image1', 'image2', 'image3']
-images = Image.open(images)
+image1, image2, image3 = 'image1', 'image2', 'image3'
+image1 = Image.open(image1)
+image2 = Image.open(image2)
+image3 = Image.open(image3)
 
-inputs = feature_extractor(images=images, return_tensors="pt")
+inputs = feature_extractor(images=[image1, image2, image3], return_tensors="pt")
 outputs = model(**inputs)
 logits = outputs.logits  # shape (batch_size, num_labels, height/4, width/4)