Update app.py

app.py

@@ -124,13 +124,15 @@ def update_ratings(state, winner_index, collection):
     update_elo_rating(collection, elo_ratings, winner, loser)
     return [('Winner: ', winner.replace('rwitz/','').replace('-lora','')), ('Loser: ', loser.replace('rwitz/','').replace('-lora',''))]
     
-def vote_up_model(state, chatbot, chatbot2, collection):
+def vote_up_model(state, chatbot, chatbot2):
+    collection = init_database()
     update_message = update_ratings(state, 0, collection)
     chatbot.append(update_message[0])
     chatbot2.append(update_message[1])
     return chatbot, chatbot2, gr.Button.update(interactive=False), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
 
-def vote_down_model(state, chatbot, chatbot2, collection):
+def vote_down_model(state, chatbot, chatbot2):
+    collection = init_database()
     update_message = update_ratings(state, 1, collection)
     chatbot2.append(update_message[0])
     chatbot.append(update_message[1])
@@ -203,9 +205,8 @@ with gr.Blocks() as demo:
         submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], queue=True)
         
         collection = init_database()
-        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1, chatbot2, collection], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
-        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2, collection], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
-        
+        upvote_btn_a.click(vote_up_model, inputs=[state, chatbot1, chatbot2], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
+        upvote_btn_b.click(vote_down_model, inputs=[state, chatbot1, chatbot2], outputs=[chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
     with gr.Tab("Leaderboard"):
         try:
             leaderboard = gr.Dataframe(refresh_leaderboard(collection))