MikeMann commited on
Commit
05862e2
·
1 Parent(s): 6075ce5

Cuda Version in Req

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