WIP

app.py

@@ -105,6 +105,10 @@ def record_feedback(output_raw, feedback, do_save=True):
         print('Commit URL:', commit_url)
         print(output_raw)
         print()
+def record_feedback_agree(output_raw, do_save=True):
+    record_feedback(output_raw, 'agree', do_save)
+def record_feedback_disagree(output_raw, do_save=True):
+    record_feedback(output_raw, 'disagree', do_save)
 
 examples = [
     # openbookqa
@@ -217,7 +221,7 @@ with gr.Blocks() as demo:
             )
     submit.click(predict, inputs=[statement, do_save], outputs=[output])
     statement.submit(predict, inputs=[statement], outputs=[output])
-    feedback_agree.click(record_feedback, inputs=[output_raw, 'agree', do_save])
-    feedback_disagree.click(record_feedback, inputs=[output_raw, 'disagree', do_save])
+    feedback_agree.click(record_feedback_agree, inputs=[output_raw, do_save])
+    feedback_disagree.click(record_feedback_disagree, inputs=[output_raw, do_save])
 
 demo.queue(concurrency_count=16).launch(debug=True)