Update app.py

app.py

@@ -413,12 +413,12 @@ with gr.Blocks(css="""
 
     async def safe_predict(image):
         try:
-            # 從 predict 函數返回 4 個結果
-            prediction, annotated_img, buttons_options, breed_detail = await predict(image)
+            # 從 predict 函數返回的值數應該與此處匹配
+            prediction, annotated_img, buttons_options = await predict(image)
             if buttons_options:  # 如果有多狗結果，顯示選項按鈕
-                return prediction, annotated_img, gr.update(visible=True, choices=buttons_options), breed_detail
+                return prediction, annotated_img, gr.update(visible=True, choices=buttons_options), gr.update(visible=True)
             else:  # 單狗預測時隱藏按鈕
-                return prediction, annotated_img, gr.update(visible=False), breed_detail
+                return prediction, annotated_img, gr.update(visible=False), gr.update(visible=False)
         except Exception as e:
             return str(e), None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)