Update app.py

app.py

@@ -54,8 +54,8 @@ with gr.Blocks() as nps:
                 gr.Markdown(f"- {cat}")
                 remove_btn = gr.Button("X", elem_id=f"remove_{i}", interactive=True)
                 remove_btn.click(
-                    fn=lambda x=cat: remove_category(x, categories),
-                    inputs=[],
+                    fn=remove_category,
+                    inputs=[gr.State(cat), category_boxes],
                     outputs=category_boxes
                 )
             category_components.append(gr.Row())