Update app.py

app.py

@@ -411,18 +411,12 @@ with gr.Blocks(css="""
     breed_buttons = gr.Radio([], label="Select breed for more details", visible=False)
     breed_details = gr.Markdown(label="Breed Details")
 
-    async def safe_predict(image):   
+    async def safe_predict(image):
         try:
-            # 從 predict 函數返回的值數應該與此處匹配
-            prediction, annotated_img, buttons_options, breed_details_visibility = await predict(image)
-            if buttons_options:  # 如果有多狗結果，顯示選項按鈕
-                return prediction, annotated_img, gr.update(visible=True, choices=buttons_options), breed_details_visibility
-            else:  # 單狗預測時隱藏按鈕
-                return prediction, annotated_img, gr.update(visible=False), breed_details_visibility
+            return await predict(image)
         except Exception as e:
             return str(e), None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
-
     input_image.change(
         safe_predict,
         inputs=input_image,