Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ rerank_models = [
|
|
| 27 |
]
|
| 28 |
|
| 29 |
collection_name = "trec_covid"
|
| 30 |
-
qdrant_url = os.getenv("
|
| 31 |
|
| 32 |
# =====================
|
| 33 |
# LOAD DATA
|
|
@@ -163,7 +163,7 @@ def evaluate_metrics(results_data, qrels_dict, k_values):
|
|
| 163 |
# =====================
|
| 164 |
|
| 165 |
def encode_and_upload():
|
| 166 |
-
client = QdrantClient(url=qdrant_url, api_key=os.getenv("
|
| 167 |
|
| 168 |
for embedding_model in embedding_models:
|
| 169 |
print(f"Encoding corpus with embedding model {embedding_model} ...")
|
|
@@ -346,7 +346,7 @@ print(f"Average relevant docs per query: {avg_relevant_docs:.2f}")
|
|
| 346 |
# --------------------
|
| 347 |
# CONFIG
|
| 348 |
# --------------------
|
| 349 |
-
QDRANT_URL = os.getenv("
|
| 350 |
EMBEDDING_MODEL = "all-MiniLM-L6-v2"
|
| 351 |
MAPPING_FILE = "int_to_doc_id.pkl"
|
| 352 |
# --------------------
|
|
|
|
| 27 |
]
|
| 28 |
|
| 29 |
collection_name = "trec_covid"
|
| 30 |
+
qdrant_url = os.getenv("QDRANT_URL", "http://localhost:6333")k_values = [1, 3, 5, 10, 20]
|
| 31 |
|
| 32 |
# =====================
|
| 33 |
# LOAD DATA
|
|
|
|
| 163 |
# =====================
|
| 164 |
|
| 165 |
def encode_and_upload():
|
| 166 |
+
client = QdrantClient(url=qdrant_url, api_key=os.getenv("QDRANT_API_KEY"))
|
| 167 |
|
| 168 |
for embedding_model in embedding_models:
|
| 169 |
print(f"Encoding corpus with embedding model {embedding_model} ...")
|
|
|
|
| 346 |
# --------------------
|
| 347 |
# CONFIG
|
| 348 |
# --------------------
|
| 349 |
+
QDRANT_URL = os.getenv("QDRANT_URL", "http://localhost:6333")COLLECTION_NAME = "trec_covid"
|
| 350 |
EMBEDDING_MODEL = "all-MiniLM-L6-v2"
|
| 351 |
MAPPING_FILE = "int_to_doc_id.pkl"
|
| 352 |
# --------------------
|