Update app.py

app.py

@@ -23,9 +23,10 @@ def visualize_segmentation(image, prompts, preds):
 
 def segment(img, clases):
     print(img)
+    image = PIL.Image.fromarray(img, 'RGB')
     prompts = clases.split(',')
 
-    inputs = processor(text=prompts, images=[img] * len(img), padding="max_length", return_tensors="pt")
+    inputs = processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
     
     outputs = model(**inputs)
     preds = outputs.logits.unsqueeze(1)