Cuda Version in Req
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -80,7 +80,7 @@ class BSIChatbot:
|
|
| 80 |
# Load existing vector store
|
| 81 |
self.vectorstore = FAISS.load_local(os.path.join(self.docs, "_embeddings"), self.embedding_model)
|
| 82 |
|
| 83 |
-
|
| 84 |
def retrieve_similar_embedding(self, query: str):
|
| 85 |
query = f"Instruct: Given a search query, retrieve the relevant passages that answer the query\nQuery:{query}"
|
| 86 |
return self.vectorstore.similarity_search(query=query, k=20)
|
|
|
|
| 80 |
# Load existing vector store
|
| 81 |
self.vectorstore = FAISS.load_local(os.path.join(self.docs, "_embeddings"), self.embedding_model)
|
| 82 |
|
| 83 |
+
#@spaces.GPU
|
| 84 |
def retrieve_similar_embedding(self, query: str):
|
| 85 |
query = f"Instruct: Given a search query, retrieve the relevant passages that answer the query\nQuery:{query}"
|
| 86 |
return self.vectorstore.similarity_search(query=query, k=20)
|
requirements.txt
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
accelerate
|
| 2 |
aiofiles
|
| 3 |
aiohappyeyeballs
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu113
|
| 2 |
accelerate
|
| 3 |
aiofiles
|
| 4 |
aiohappyeyeballs
|