Update app.py

app.py

@@ -21,7 +21,7 @@ def query(payload):
 
 #-----------------------------------------------------------
 
-API_URL_evidence = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
+API_URL_evidence = "https://api-inference.huggingface.co/models/t5-3b"
 headers_evidence = {"Authorization": HF_SPACES_API_KEY}
 
 def query_evidence(payload):
@@ -102,11 +102,11 @@ claim_text=st.text_area("Enter your claim:", on_change=proc, key='text_key')
 # if evidence_text:
 #     st.write("You entered: ", evidence_text)
         
-# evidence_text=st.text_area("Enter your evidence:")
+evidence_text=st.text_area("Enter your evidence:")
 
-form_evidence = st.form(key='my_evidence')
-form_evidence.text_input(label='Enter your evidence')
-evidence_text = form_evidence.form_submit_button(label='Submit')
+# form_evidence = st.form(key='my_evidence')
+# form_evidence.text_input(label='Enter your evidence')
+# evidence_text = form_evidence.form_submit_button(label='Submit')
 
 if evidence_text:
     st.caption(':green[Kindly hold on for a few minutes while the QA pairs are being generated]')
@@ -384,7 +384,7 @@ def gen_qa_who(df):
                     list_of_ans_who.append(f"""Ans{j+1}:{answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     #time.sleep(10)
-                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['translation_text']
                     if answer_evidence.lower() in evidence.lower():
                         list_of_evidence_answer_who.append(f"""Evidence{j+1}:{answer_evidence}""")
                     else:
@@ -444,7 +444,7 @@ def gen_qa_what(df):
                     list_of_ans_what.append(f"""Ans{j+1}:{answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     #time.sleep(10)
-                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['translation_text']
                     if answer_evidence.lower() in evidence.lower():
                         list_of_evidence_answer_what.append(f"""Evidence{j+1}:{answer_evidence}""")
                         
@@ -506,7 +506,7 @@ def gen_qa_why(df):
                     list_of_ans_why.append(f"""Ans{j+1}:{answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     #time.sleep(10)
-                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['translation_text']
                     if answer_evidence.lower() in evidence.lower():
                         list_of_evidence_answer_why.append(f"""Evidence{j+1}:{answer_evidence}""")
                     else:
@@ -567,7 +567,7 @@ def gen_qa_when(df):
                     list_of_ans_when.append(f"""Ans{j+1}:{answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     #time.sleep(10)
-                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['translation_text']
                     if answer_evidence.lower() in evidence.lower():
                         list_of_evidence_answer_when.append(f"""Evidence{j+1}:{answer_evidence}""")
                     else:
@@ -628,7 +628,7 @@ def gen_qa_where(df):
                     list_of_ans_where.append(f"""Ans{j+1}:{answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     #time.sleep(10)
-                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['translation_text']
                     if answer_evidence.lower() in evidence.lower():
                         list_of_evidence_answer_where.append(f"""Evidence{j+1}:{answer_evidence}""")
                     else: