Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -297,7 +297,7 @@ async def predict(image):
|
|
297 |
|
298 |
|
299 |
for i, (cropped_image, detection_confidence, box) in enumerate(dogs):
|
300 |
-
buttons_html = ""
|
301 |
top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
|
302 |
color = color_list[i % len(color_list)]
|
303 |
draw.rectangle(box, outline=color, width=3)
|
@@ -313,17 +313,15 @@ async def predict(image):
|
|
313 |
dogs_info += format_description_html(description, breed)
|
314 |
|
315 |
elif combined_confidence >= 0.15:
|
316 |
-
#
|
317 |
dogs_info += f"<p>Top 3 possible breeds:</p><ul>"
|
318 |
for j, (breed, prob) in enumerate(zip(topk_breeds[:3], topk_probs_percent[:3])):
|
319 |
prob = float(prob.replace('%', ''))
|
320 |
-
|
321 |
-
|
322 |
-
# ๅจๆฏๅๅ็จฎไนๅพๆทปๅ ๆ้
|
323 |
button_id = f"Dog {i+1}: More about {breed}"
|
324 |
-
dogs_info += f
|
|
|
325 |
buttons.append(button_id)
|
326 |
-
|
327 |
dogs_info += "</ul>"
|
328 |
|
329 |
else:
|
@@ -332,6 +330,7 @@ async def predict(image):
|
|
332 |
dogs_info += '</div>' # ็ตๆ็ถๅ็็่ณ่จๅๅก
|
333 |
|
334 |
|
|
|
335 |
|
336 |
buttons_html = ""
|
337 |
|
|
|
297 |
|
298 |
|
299 |
for i, (cropped_image, detection_confidence, box) in enumerate(dogs):
|
300 |
+
buttons_html = ""
|
301 |
top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
|
302 |
color = color_list[i % len(color_list)]
|
303 |
draw.rectangle(box, outline=color, width=3)
|
|
|
313 |
dogs_info += format_description_html(description, breed)
|
314 |
|
315 |
elif combined_confidence >= 0.15:
|
316 |
+
# ๅๅบๅไธๅๅฏ่ฝ็ๅ็จฎ๏ผไธฆๅจๆฏๅๅ็จฎๆ็ๆๆ้
|
317 |
dogs_info += f"<p>Top 3 possible breeds:</p><ul>"
|
318 |
for j, (breed, prob) in enumerate(zip(topk_breeds[:3], topk_probs_percent[:3])):
|
319 |
prob = float(prob.replace('%', ''))
|
320 |
+
# ๆฏๅๅ็จฎๅ็จฑๅพ้ข็ซๅณ็ๆๆ้
|
|
|
|
|
321 |
button_id = f"Dog {i+1}: More about {breed}"
|
322 |
+
dogs_info += f"<li><strong>{breed}</strong> ({prob:.2f}% confidence)</li>"
|
323 |
+
dogs_info += f'<button class="breed-button" style="margin-left: 10px;" onclick="handle_button_click(\'{button_id}\')">{breed}</button>'
|
324 |
buttons.append(button_id)
|
|
|
325 |
dogs_info += "</ul>"
|
326 |
|
327 |
else:
|
|
|
330 |
dogs_info += '</div>' # ็ตๆ็ถๅ็็่ณ่จๅๅก
|
331 |
|
332 |
|
333 |
+
|
334 |
|
335 |
buttons_html = ""
|
336 |
|