Update app.py

app.py

@@ -75,7 +75,7 @@ def retrieve_context(query, contexts):
 
 
 # Define your function to generate answers
-def generate_answer(question, context, ground):
+def generate_answer(question, context, title, ground):
     contexts = chunk_splitter(context)
     context = retrieve_context(question, contexts)
     
@@ -108,7 +108,8 @@ def list_examples():
         context = example["article"]
         question = example["question"]
         answer = example["answer"]
-        examples.append([question, context, answer])
+        title = example["title"]
+        examples.append([question, context, title, answer])
     return examples
 
 
@@ -118,6 +119,7 @@ iface = gr.Interface(
     inputs=[
         Textbox(label="Question"),
         Textbox(label="Context"),
+        Textbox(label="Article title"),
         Textbox(label="Ground truth")
     ],
     outputs=[