Update app.py

app.py

@@ -66,12 +66,11 @@ llm = GroqLLM(api_key=groq_api_key, model_name="deepseek-r1-distill-llama-70b")
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.documents import Document
 from typing import List
+from pydantic import Field
 
 class SimpleRetriever(BaseRetriever):
-    def __init__(self, documents: List[Document], embeddings: List):
-        super().__init__()
-        self.documents = documents
-        self.embeddings = embeddings
+    documents: List[Document] = Field(...)
+    embeddings: List = Field(...)
 
     def _get_relevant_documents(self, query: str) -> List[Document]:
         inputs = tokenizer(query, return_tensors="pt", padding=True, truncation=True)