Update app.py

app.py

@@ -16,7 +16,9 @@ class SamAutomaticMaskGenerator:
         self.sam_pipeline = sam_pipeline
 
     def generate(self, image_rgb):
-        outputs = self.sam_pipeline(image_rgb, points_per_batch=32)
+        # Convertir el array de NumPy a PIL Image
+        image_pil = Image.fromarray(image_rgb)
+        outputs = self.sam_pipeline(image_pil, points_per_batch=32)
         mask = np.array(outputs['masks'], dtype=np.uint8)
         return mask