Update app.py

app.py

@@ -518,8 +518,11 @@ async def predict(image):
         draw = ImageDraw.Draw(annotated_image)
         font = ImageFont.load_default()
 
-        dogs_info = ""
-        buttons_html = ""
+        # dogs_info = ""
+        # buttons_html = ""
+
+        dogs_info += f'<div class="dog-info" style="border-left: 5px solid {color}; margin-bottom: 20px; padding: 15px;">'
+        dogs_info += f'<h2>Dog {i+1}</h2>'
 
         for i, (cropped_image, detection_confidence, box) in enumerate(dogs):
             top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)