Update app.py

app.py

@@ -528,16 +528,16 @@ def main(device, segment_type):
 
                 if (pipe.tokenizer("man")["input_ids"][1] in pipe.tokenizer(args.prompt)["input_ids"][1:-1]) and (
                         pipe.tokenizer("woman")["input_ids"][1] in pipe.tokenizer(args.prompt)["input_ids"][1:-1]):
-                    mask1, mask2 = predict_mask(detect_model, sam, image[0], ['man', 'woman'], args.segment_type, confidence=0.3,
+                    mask1, mask2 = predict_mask(detect_model, sam, image[0], ['man', 'woman'], args.segment_type, confidence=0.05,
                                  threshold=0.5)
 
                 elif pipe.tokenizer("man")["input_ids"][1] in pipe.tokenizer(args.prompt)["input_ids"][1:-1]:
-                    mask1 = predict_mask(detect_model, sam, image[0], ['man'], args.segment_type, confidence=0.3,
+                    mask1 = predict_mask(detect_model, sam, image[0], ['man'], args.segment_type, confidence=0.05,
                                          threshold=0.5)
                     mask2 = None
 
                 elif pipe.tokenizer("woman")["input_ids"][1] in pipe.tokenizer(args.prompt)["input_ids"][1:-1]:
-                    mask2 = predict_mask(detect_model, sam, image[0], ['woman'], args.segment_type, confidence=0.3,
+                    mask2 = predict_mask(detect_model, sam, image[0], ['woman'], args.segment_type, confidence=0.05,
                                          threshold=0.5)
                     mask1 = None
                 else: