Parth211 commited on
Commit
cb48794
·
verified ·
1 Parent(s): 9a39944
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -358,7 +358,7 @@ def conversation(qa_chain, message, history, gpt2_model, gpt2_tokenizer, bias_pi
358
  new_history = history + [(message, response_answer)]
359
 
360
  # Evaluate the metrics
361
- metrics = evaluate_all(question_by_user, answer_of_question, context)
362
  evaluation_metrics = display_metrics(metrics)
363
 
364
  return (qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page,
@@ -465,12 +465,12 @@ def demo():
465
 
466
  # Chatbot events
467
  msg.submit(conversation, \
468
- inputs=[qa_chain, msg, chatbot], \
469
  outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page,metrics_output], \
470
  queue=False)
471
 
472
  submit_btn.click(conversation,
473
- inputs=[qa_chain, msg, history],
474
  outputs=[qa_chain, chatbot, history, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page, metrics_output])
475
 
476
  clear_btn.click(lambda: [None, "", 0, "", 0, "", 0],
 
358
  new_history = history + [(message, response_answer)]
359
 
360
  # Evaluate the metrics
361
+ metrics = evaluate_all(question_by_user, answer_of_question, context,gpt2_model, gpt2_tokenizer, bias_pipeline)
362
  evaluation_metrics = display_metrics(metrics)
363
 
364
  return (qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page,
 
465
 
466
  # Chatbot events
467
  msg.submit(conversation, \
468
+ inputs=[qa_chain, msg, chatbot, gpt2_model, gpt2_tokenizer, bias_pipeline], \
469
  outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page,metrics_output], \
470
  queue=False)
471
 
472
  submit_btn.click(conversation,
473
+ inputs=[qa_chain, msg, history, gpt2_model, gpt2_tokenizer, bias_pipeline],
474
  outputs=[qa_chain, chatbot, history, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page, metrics_output])
475
 
476
  clear_btn.click(lambda: [None, "", 0, "", 0, "", 0],