Update app.py

app.py

@@ -21,7 +21,7 @@ def query(payload):
 
 #-----------------------------------------------------------
 
-API_URL_evidence = "https://api-inference.huggingface.co/models/t5-3b"
+API_URL_evidence = "https://api-inference.huggingface.co/models/google/flan-t5-xxl"
 headers_evidence = {"Authorization": HF_SPACES_API_KEY}
 
 def query_evidence(payload):
@@ -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]['translation_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_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]['translation_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_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]['translation_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_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]['translation_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_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]['translation_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
                     if answer_evidence.lower() in evidence.lower():
                         list_of_evidence_answer_where.append(f"""Evidence{j+1}:{answer_evidence}""")
                     else: