Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -244,7 +244,7 @@ def _predict_single_dog(image):
|
|
244 |
# return error_msg, None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), None
|
245 |
|
246 |
|
247 |
-
async def detect_multiple_dogs(image, conf_threshold=0.
|
248 |
results = model_yolo(image, conf=conf_threshold, iou=iou_threshold)[0]
|
249 |
dogs = []
|
250 |
|
|
|
244 |
# return error_msg, None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), None
|
245 |
|
246 |
|
247 |
+
async def detect_multiple_dogs(image, conf_threshold=0.25, iou_threshold=0.4, merge_threshold=0.5):
|
248 |
results = model_yolo(image, conf=conf_threshold, iou=iou_threshold)[0]
|
249 |
dogs = []
|
250 |
|