Update app.py

app.py

@@ -19,7 +19,7 @@ def clear_chat(state):
 
 
 
-from datasets import load_dataset
+from datasets import load_dataset,DatasetDict
 import requests
 import os
 
@@ -40,6 +40,7 @@ def update_elo_rating(new_rating):
     updated_dataset = elo_ratings.map(update_rating)
 
     # Convert updated dataset to a dictionary for pushing
+    updated_dataset=DatasetDict({'train':updated_dataset})
     updated_dataset.push_to_hub("rwitz/mistral-elo-ratings",token=os.environ.get("huggingface_token"))
     if response.status_code == 200:
         print("Successfully updated the dataset")