Update app.py

app.py

@@ -24,7 +24,7 @@ def visualize_segmentation(image, prompts, preds):
 def segment(img, clases):
     prompts = clases.split(',')
 
-    inputs = processor(text=prompts, images=[image] * len(img), padding="max_length", return_tensors="pt")
+    inputs = processor(text=prompts, images=[img] * len(img), padding="max_length", return_tensors="pt")
     
     with torch.no_grad():
       outputs = model(**inputs)