Update app.py
Browse files
app.py
CHANGED
@@ -74,10 +74,6 @@ class SimpleRetriever(BaseRetriever):
|
|
74 |
documents: List[Document] = field(default_factory=list)
|
75 |
embeddings: List = field(default_factory=list)
|
76 |
|
77 |
-
def __init__(self):
|
78 |
-
super().__init__()
|
79 |
-
self.documents, self.embeddings = build_pdf_index()
|
80 |
-
|
81 |
def _get_relevant_documents(self, query: str) -> List[Document]:
|
82 |
inputs = tokenizer(query, return_tensors="pt", padding=True, truncation=True)
|
83 |
with torch.no_grad():
|
|
|
74 |
documents: List[Document] = field(default_factory=list)
|
75 |
embeddings: List = field(default_factory=list)
|
76 |
|
|
|
|
|
|
|
|
|
77 |
def _get_relevant_documents(self, query: str) -> List[Document]:
|
78 |
inputs = tokenizer(query, return_tensors="pt", padding=True, truncation=True)
|
79 |
with torch.no_grad():
|