Update app.py

app.py

@@ -23,8 +23,8 @@ def find_similarity(base64_image, text_input):
         # Decode the base64 image to bytes
         image_bytes = base64.b64decode(base64_image)
 
-        # Convert the bytes to a PIL image
-        image = Image.open(BytesIO(image_bytes))
+        # Convert the bytes to a numpy array
+        image = np.array(Image.open(BytesIO(image_bytes)))
 
         # Preprocess the image
         image = preprocess(image).unsqueeze(0).to(device)
@@ -55,7 +55,7 @@ def segment_image(input_image, text_input):
         image = image.convert("RGB")
 
         mask_generator = SamAutomaticMaskGenerator(sam)
-        masks = mask_generator.generate(image)
+        masks = mask_generator.generate(np.array(image))
 
         segmented_regions = []  # List to store segmented regions with similarity scores