opex792 commited on
Commit
9426abb
·
verified ·
1 Parent(s): fa3df7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=20):
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 * 1.1)))
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)} предварительных результатов поиска.")