Update app.py

app.py

@@ -28,7 +28,7 @@ def get_user_elo_ratings():
     elo_ratings = dataset['train']  # or the relevant split
     return elo_ratings
 
-def update_elo_rating(new_rating):
+def update_elo_rating(new_rating,winner,loser):
     # Fetch the current dataset
     elo_ratings = get_user_elo_ratings()
 
@@ -36,6 +36,7 @@ def update_elo_rating(new_rating):
 
     # Update the dataset
     # Convert updated dataset to a dictionary for pushing
+    updated_ratings=update_elo_ratings(elo_ratings,winner,loser)
     elo_ratings.push_to_hub("rwitz/mistral-elo-ratings",token=os.environ.get("huggingface_token"))
     
 # Function to get bot response
@@ -97,7 +98,7 @@ def update_ratings(state, winner_index):
     loser = state['last_bots'][1 - winner_index]
     
     elo_ratings = update_elo_ratings(elo_ratings, winner, loser)
-    update_elo_rating(elo_ratings)
+    update_elo_rating(elo_ratings,winner,loser)
     return [('Winner: ',state['last_bots'][winner_index]),('Loser: ',state['last_bots'][1 - winner_index])]
 
 def vote_up_model(state, chatbot,chatbot2):