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)} предварительных результатов поиска.")
|