Update app.py

app.py

@@ -21,6 +21,10 @@ with gr.Blocks() as demo:
   user_id = gr.Textbox(placeholder="Enter user id for personalization: ")
   with gr.Column():
     for arxiv_paper in search.results():
+      temp_id = arxiv_paper.entry_id
+      temp_id = temp_id.split("/")[-1]
+      temp_id = temp_id.replace(".", "").replace("v1", "")
+      temp_id = int(temp_id)
       with gr.Column():
         with gr.Column():
           gr.Markdown("<center><h3>" + arxiv_paper.title + "</h3></center>")
@@ -28,9 +32,9 @@ with gr.Blocks() as demo:
         with gr.Row():
           more_button = gr.Button("More like this! 😎")
           # add user id value later
-          more_button.click(hf_data_upload(0, arxiv_paper.title, 1))
+          more_button.click(hf_data_upload(0, temp_id, 1))
           #button.click(flip_image, inputs=image_input, outputs=image_output)
           less_button = gr.Button("Less like this! 😕")
-          less_button.click(hf_data_upload(0, arxiv_paper.title, 0))
+          less_button.click(hf_data_upload(0, temp_id, 0))
 
 demo.launch()