Update app.py

app.py

@@ -172,16 +172,19 @@ def predict(image):
     try:
         # 使用 YOLO 偵測狗
         results = model_yolo(image)
-        dogs = results.xyxy[0]  # 提取偵測到的狗的邊界框
-
-        if len(dogs) == 0:
+        
+        # 檢查 YOLO 輸出
+        boxes = results[0].boxes  # 修改這裡，使用 results[0].boxes 來提取邊界框
+        
+        if len(boxes) == 0:
             return "No dog detected in the image.", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
         explanations = []
         visible_buttons = []
 
-        for i, box in enumerate(dogs):
-            x1, y1, x2, y2 = map(int, box[:4])
+        for i, box in enumerate(boxes):
+            # 提取每隻狗的區域
+            x1, y1, x2, y2 = map(int, box.xyxy[0])  # 使用 box.xyxy 來提取邊界框座標
             cropped_image = image.crop((x1, y1, x2, y2))  # 裁剪狗區域
             image_tensor = preprocess_image(cropped_image)
 
@@ -222,6 +225,7 @@ def predict(image):
         return f"An error occurred: {e}", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
 
+
 def format_description(description, breed):
     if isinstance(description, dict):
         formatted_description = "\n\n".join([f"**{key}**: {value}" for key, value in description.items()])