Update app.py

app.py

@@ -12,7 +12,8 @@ search = arxiv.Search(
 
 def hf_data_upload(user_id, paper_id, feedback):
   new_row = {"user_id": user_id, "paper_id": paper_id, "feedback": feedback}
-  ds = load_dataset("CShorten/Last-Week-on-ArXiv").add_item(new_row)
+  ds = load_dataset("CShorten/Last-Week-on-ArXiv")["train"]
+  ds = ds.add_item(new_row)
   ds.push_to_hub("CShorten/Last-Week-on-ArXiv") 
 
 with gr.Blocks() as demo: