Update app.py

app.py

@@ -18,11 +18,12 @@ class BasicAgent:
         # Load the BERT model fine-tuned on the SQuAD dataset
         self.qa_pipeline = pipeline("question-answering", model="bert-large-uncased-whole-word-masking-finetuned-squad")
     
-    def __call__(self, question: str, context: str) -> str:
+    def __call__(self, question: str) -> str:
         """
         Process the question using the BERT model fine-tuned on SQuAD, and return an answer based on the context.
         """
         try:
+            context = f"answer the following question as briefly as possible{question}"
             # Use the QA pipeline to get an answer based on the context and the question
             result = self.qa_pipeline(question=question, context=context)
             answer = result["answer"]