Update app.py

app.py

@@ -271,7 +271,7 @@ css = """
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown(description)
-    vote_id = gr.State(0)
+    # vote_id = gr.State(0)
     with gr.Tab(label="Protein caption"):
         with gr.Row():
             with gr.Column():
@@ -291,7 +291,6 @@ with gr.Blocks(css=css) as demo:
             feedback_markdown = gr.Markdown(label="Output")
         with gr.Row():
             feedback_btn = gr.Button(value="Feedback Submit", elem_classes="feedback")
-        # O14813 train index 127, 266, 738, 1060 test index 4
         gr.Examples(
             examples=[
                 ["MDYSYLNSYDSCVAAMEASAYGDFGACSQPGGFQYSPLRPAFPAAGPPCPALGSSNCALGALRDHQPAPYSAVPYKFFPEPSGLHEKRKQRRIRTTFTSAQLKELERVFAETHYPDIYTREELALKIDLTEARVQVWFQNRRAKFRKQERAASAKGAAGAAGAKKGEARCSSEDDDSKESTCSPTPDSTASLPPPPAPGLASPRLSPSPLPVALGSGPGPGPGPQPLKGALWAGVAGGGGGGPGAGAAELLKAWQPAESGPGPFSGVLSSFHRKPGPALKTNLF", ''],
@@ -308,8 +307,8 @@ with gr.Blocks(css=css) as demo:
             label='Try examples'
         )
         submit_btn.click(generate_caption, [input_protein, prompt], [output_markdown])
-        upvote_button.click(upvote, vote_id, vote_markdown)
-        downvote_button.click(downvote, vote_id, vote_markdown)
+        upvote_button.click(upvote, input_protein, vote_markdown)
+        downvote_button.click(downvote, input_protein, vote_markdown)
         feedback_btn.click(save_feedback, [inputs], [feedback_markdown])
 
 demo.launch(debug=True)