Update app.py
Browse files
app.py
CHANGED
|
@@ -344,7 +344,7 @@ def rerank_with_api(query, results, top_k):
|
|
| 344 |
logging.error(f"Ошибка при запросе к API реранжировщика: {e}")
|
| 345 |
return []
|
| 346 |
|
| 347 |
-
def search_movies(query, top_k=
|
| 348 |
"""Выполняет поиск фильмов по запросу."""
|
| 349 |
global search_in_progress
|
| 350 |
search_in_progress = True
|
|
@@ -387,7 +387,7 @@ def search_movies(query, top_k=20):
|
|
| 387 |
FROM {embeddings_table} m, query_embedding
|
| 388 |
ORDER BY similarity DESC
|
| 389 |
LIMIT %s
|
| 390 |
-
""", (query_crc32, int(top_k *
|
| 391 |
|
| 392 |
results = cur.fetchall()
|
| 393 |
logging.info(f"Найдено {len(results)} предварительных результатов поиска.")
|
|
|
|
| 344 |
logging.error(f"Ошибка при запросе к API реранжировщика: {e}")
|
| 345 |
return []
|
| 346 |
|
| 347 |
+
def search_movies(query, top_k=30):
|
| 348 |
"""Выполняет поиск фильмов по запросу."""
|
| 349 |
global search_in_progress
|
| 350 |
search_in_progress = True
|
|
|
|
| 387 |
FROM {embeddings_table} m, query_embedding
|
| 388 |
ORDER BY similarity DESC
|
| 389 |
LIMIT %s
|
| 390 |
+
""", (query_crc32, int(top_k * 2)))
|
| 391 |
|
| 392 |
results = cur.fetchall()
|
| 393 |
logging.info(f"Найдено {len(results)} предварительных результатов поиска.")
|