ktllc commited on
Commit
de97c4c
·
1 Parent(s): 9a9faba

Update app.py

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