ktllc commited on
Commit
d65cf39
·
1 Parent(s): cf83b7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ def segment_image(input_image, text_input):
54
  image_bytes = base64.b64decode(input_image)
55
  image = Image.open(BytesIO(image_bytes))
56
 
57
- image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
58
  mask_generator = SamAutomaticMaskGenerator(sam)
59
  masks = mask_generator.generate(image)
60
 
 
54
  image_bytes = base64.b64decode(input_image)
55
  image = Image.open(BytesIO(image_bytes))
56
 
57
+ image = cv2.cvtColor(np.array(image), cv2.COLOR_BGR2RGB)
58
  mask_generator = SamAutomaticMaskGenerator(sam)
59
  masks = mask_generator.generate(image)
60