Daniel Doña commited on
Commit
1796e7a
·
1 Parent(s): 3df2dd0

Remove GPU for semantic models

Browse files
Files changed (1) hide show
  1. src/SemanticSearch.py +2 -3
src/SemanticSearch.py CHANGED
@@ -7,7 +7,6 @@ import requests
7
  import itertools
8
  import torch
9
 
10
- import spaces
11
 
12
  import chromadb
13
 
@@ -201,8 +200,8 @@ class SemanticSearch:
201
 
202
  self.collection = client.get_collection(name=self.collection_name)
203
 
204
- @spaces.GPU
205
- def extract(self, nlq: str, n_results:int=10, n_candidates:int=100, str_match:bool=True, rerank:bool=True):
206
 
207
  embedding = self.get_text_embeddings_local([nlq])[0].tolist()
208
 
 
7
  import itertools
8
  import torch
9
 
 
10
 
11
  import chromadb
12
 
 
200
 
201
  self.collection = client.get_collection(name=self.collection_name)
202
 
203
+
204
+ def extract(self, nlq: str, n_results:int=10, n_candidates:int=50, str_match:bool=True, rerank:bool=True):
205
 
206
  embedding = self.get_text_embeddings_local([nlq])[0].tolist()
207