M17idd commited on
Commit
f8d1cfb
·
verified ·
1 Parent(s): f21cdf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
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():