Update app.py

app.py

@@ -438,27 +438,27 @@ async def predict(image):
                 "html_output": html_output
             }
             return html_output, annotated_image, gr.update(visible=True, choices=all_buttons), initial_state
-        else:
-            initial_state = {
+         else:      
+             initial_state = {
                 "dogs_info": ''.join(dogs_info),
                 "buttons": [],
                 "show_back": False,
                 "image": annotated_image,
                 "is_multi_dog": len(dogs) > 1,
                 "html_output": html_output
-            }
-            return html_output, annotated_image, gr.update(visible=False, choices=[]), initial_state
+             }
+             return html_output, annotated_image, gr.update(visible=False, choices=[]), initial_state
 
-        else:
-            initial_state = {
+         else:
+             initial_state = {
                 "dogs_info": dogs_info,
                 "buttons": [],
                 "show_back": False,
                 "image": annotated_image,
                 "is_multi_dog": len(dogs) > 1,
                 "html_output": html_output
-            }
-            return html_output, annotated_image, gr.update(visible=False, choices=[]), initial_state
+             }
+             return html_output, annotated_image, gr.update(visible=False, choices=[]), initial_state
 
     except Exception as e:
         error_msg = f"An error occurred: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"