Update app.py

app.py

@@ -135,8 +135,7 @@ class RAG:
       return answer
 
     def extractive_query(self, question):
-      context = self.retriever.retrieve_top_k(question, k=15)
-      
+      context = self.retriever.retrieve_top_k(question, k=4)
       
       inputs = self.generator_tokenizer(question, ". ".join(context), return_tensors="pt", truncation=True, max_length=150, padding="max_length")
       with torch.no_grad():