Update app.py

app.py

@@ -72,12 +72,11 @@ with gr.Blocks() as demo:
             info_output = gr.Textbox(label="Image Information", lines=10)
     
     def update_slider(category_id):
-        _, _, max_index = draw_annotations(0, category_id)
-        return gr.Slider.update(maximum=max_index)
+        img, info, max_index = draw_annotations(0, category_id)
+        return gr.Slider.update(maximum=max_index), img, info
 
     # Update image and info when slider or category changes
-    category_id_dropdown.change(update_slider, inputs=category_id_dropdown, outputs=image_index)
-    category_id_dropdown.change(draw_annotations, inputs=[image_index, category_id_dropdown], outputs=[image_output, info_output, image_index])
+    category_id_dropdown.change(update_slider, inputs=category_id_dropdown, outputs=[image_index, image_output, info_output])
     image_index.change(draw_annotations, inputs=[image_index, category_id_dropdown], outputs=[image_output, info_output, image_index])
 
     # Display initial image and info