sigyllly commited on
Commit
c11619c
1 Parent(s): 6e7b1c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def process_image(image, prompt):
27
  return mask
28
 
29
  def get_masks(prompts, img, threshold):
30
- prompts = prompts.split(",")
31
  masks = []
32
  for prompt in prompts:
33
  mask = process_image(img, prompt)
 
27
  return mask
28
 
29
  def get_masks(prompts, img, threshold):
30
+ prompts = [p.strip() for p in prompts.split(",")]
31
  masks = []
32
  for prompt in prompts:
33
  mask = process_image(img, prompt)