Update app.py

app.py

@@ -129,10 +129,11 @@ def create_demo():
         for model_name, _, _, vote_button, vote_count in response_boxes:
             vote_button.click(
                 lambda votes, name=model_name: handle_votes(votes, name),
-                inputs=[vote_counts],
-                outputs=[vote_counts],
+                inputs=[vote_state],
+                outputs=[vote_state, vote_count],  # Update both the state and the UI element
             )
 
+
         # Update model visibility when the model selection changes
         selected_models.change(
             update_model_visibility,