Update app.py

app.py

@@ -138,7 +138,7 @@ def retrieve_context(question, contexts):
 
 
 # Define your function to generate answers
-def generate_answer(question, context):
+def generate_answer(question, context, ground):
     contexts = chunk_splitter(clean_data(context))
     context = retrieve_context(question, contexts)
     
@@ -164,7 +164,7 @@ def generate_answer(question, context):
     # Paraphrase answer
     paraphrased_answer = paraphrase_answer(question, generated_answer)
 
-    return generated_answer, context, paraphrased_answer
+    return generated_answer, context, paraphrased_answer, ground
 
 
 # Define a function to list examples from the dataset
@@ -183,12 +183,14 @@ iface = gr.Interface(
     fn=generate_answer,
     inputs=[
         Textbox(label="Question"),
-        Textbox(label="Context")
+        Textbox(label="Context"),
+        Textbox(label="Ground truth")
     ],
     outputs=[
         Textbox(label="Generated Answer"),
         Textbox(label="Retrieved Context"),
-        Textbox(label="Natural Answer"),
+        Textbox(label="Natural Answer")
+        Textbox(label="Ground Truth"),
     ],
     examples=list_examples()
 )