Spaces:
Runtime error
Runtime error
Daniel Doña
commited on
Commit
·
1796e7a
1
Parent(s):
3df2dd0
Remove GPU for semantic models
Browse files- 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 |
-
|
| 205 |
-
def extract(self, nlq: str, n_results:int=10, n_candidates:int=
|
| 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 |
|