Spaces:
Running
Running
changed embedder to paraphrase-multilingual-miniLM-L12-v2
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from sentence_transformers import SentenceTransformer
|
|
| 6 |
import time
|
| 7 |
|
| 8 |
# --- SETUP ---
|
| 9 |
-
model = SentenceTransformer("
|
| 10 |
|
| 11 |
QDRANT_URL = os.environ.get("QDRANT_URL")
|
| 12 |
QDRANT_API_KEY = os.environ.get("QDRANT_API_KEY")
|
|
@@ -143,7 +143,7 @@ def search_free_text_animated(query_text, max_results):
|
|
| 143 |
return
|
| 144 |
|
| 145 |
try:
|
| 146 |
-
vec = model.encode(f"
|
| 147 |
hits_response = client.query_points(
|
| 148 |
collection_name=COLLECTION_NAME,
|
| 149 |
query=vec,
|
|
|
|
| 6 |
import time
|
| 7 |
|
| 8 |
# --- SETUP ---
|
| 9 |
+
model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", device="cpu")
|
| 10 |
|
| 11 |
QDRANT_URL = os.environ.get("QDRANT_URL")
|
| 12 |
QDRANT_API_KEY = os.environ.get("QDRANT_API_KEY")
|
|
|
|
| 143 |
return
|
| 144 |
|
| 145 |
try:
|
| 146 |
+
vec = model.encode(f"{query_text}")[:VECTOR_SIZE]
|
| 147 |
hits_response = client.query_points(
|
| 148 |
collection_name=COLLECTION_NAME,
|
| 149 |
query=vec,
|