Update app.py

app.py

@@ -10,7 +10,7 @@ processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined")
 model = CLIPSegForImageSegmentation.from_pretrained("CIDAS/clipseg-rd64-refined")
 
 
-def process_image(image, prompt):
+def process_image(image, prompt, threhsold):
     inputs = processor(
         text=prompt, images=image, padding="max_length", return_tensors="pt"
     )
@@ -45,7 +45,7 @@ def get_masks(prompts, img, threhsold):
     prompts = prompts.split(",")
     masks = []
     for prompt in prompts:
-        mask = process_image(img, prompt)
+        mask = process_image(img, prompt, threhsold)
         mask = mask > threhsold
         masks.append(mask)
     return masks