Spaces:
Running
Running
remove top level retriver model
Browse files
backend/semantic_search.py
CHANGED
|
@@ -11,7 +11,7 @@ VECTOR_COLUMN = os.getenv("VECTOR_COLUMN", "vector")
|
|
| 11 |
TEXT_COLUMN = os.getenv("TEXT_COLUMN", "text")
|
| 12 |
BATCH_SIZE = int(os.getenv("BATCH_SIZE", 32))
|
| 13 |
|
| 14 |
-
retriever = SentenceTransformer(os.getenv("EMB_MODEL"))
|
| 15 |
|
| 16 |
|
| 17 |
def retrieve(query, k, table_name, embedding_model_name):
|
|
|
|
| 11 |
TEXT_COLUMN = os.getenv("TEXT_COLUMN", "text")
|
| 12 |
BATCH_SIZE = int(os.getenv("BATCH_SIZE", 32))
|
| 13 |
|
| 14 |
+
#retriever = SentenceTransformer(os.getenv("EMB_MODEL"))
|
| 15 |
|
| 16 |
|
| 17 |
def retrieve(query, k, table_name, embedding_model_name):
|