Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -221,7 +221,7 @@ def calculate_iou(box1, box2):
|
|
221 |
|
222 |
async def process_single_dog(image):
|
223 |
top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(image)
|
224 |
-
if top1_prob < 0.
|
225 |
initial_state = {
|
226 |
"explanation": "The image is unclear or the breed is not in the dataset. Please upload a clearer image of a dog.",
|
227 |
"buttons": [],
|
@@ -234,7 +234,7 @@ async def process_single_dog(image):
|
|
234 |
breed = topk_breeds[0]
|
235 |
description = get_dog_description(breed)
|
236 |
|
237 |
-
if top1_prob >= 0.
|
238 |
formatted_description = format_description(description, breed)
|
239 |
initial_state = {
|
240 |
"explanation": formatted_description,
|
|
|
221 |
|
222 |
async def process_single_dog(image):
|
223 |
top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(image)
|
224 |
+
if top1_prob < 0.15:
|
225 |
initial_state = {
|
226 |
"explanation": "The image is unclear or the breed is not in the dataset. Please upload a clearer image of a dog.",
|
227 |
"buttons": [],
|
|
|
234 |
breed = topk_breeds[0]
|
235 |
description = get_dog_description(breed)
|
236 |
|
237 |
+
if top1_prob >= 0.45:
|
238 |
formatted_description = format_description(description, breed)
|
239 |
initial_state = {
|
240 |
"explanation": formatted_description,
|