Update app.py
Browse files
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)
|