Update app.py

app.py

@@ -144,11 +144,11 @@ def retrieve_context(question, contexts):
 # Define your function to generate answers
 def generate_answer(question, context, ground, do_pretrained, do_natural, do_pretrained_natural):
     contexts = chunk_splitter(clean_data(context))
-    context = retrieve_context(question, contexts)
-    ground_in_context = ground in context
+    retrieved_context = retrieve_context(question, contexts)
+    ground_in_context = ground in retrieved_context
     
     # Combine question and context
-    input_text = f"question: {question} context: {context}"
+    input_text = f"question: {question} context: {retrieved_context}"
 
     # Tokenize the input text
     input_ids = tokenizer(
@@ -183,7 +183,7 @@ def generate_answer(question, context, ground, do_pretrained, do_natural, do_pre
     if do_pretrained_natural:
         pretrained_paraphrased_answer = paraphrase_answer(question, generated_answer, True)
 
-    return generated_answer, paraphrased_answer, ground_in_context, pretrained_answer, pretrained_paraphrased_answer, context
+    return generated_answer, paraphrased_answer, ground_in_context, pretrained_answer, pretrained_paraphrased_answer, retrieved_context
 
 
 # Define a function to list examples from the dataset