opex792 commited on
Commit
b7a12de
·
verified ·
1 Parent(s): bcb5bfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -216,7 +216,7 @@ def process_batch(batch):
216
 
217
  try:
218
  for movie in batch:
219
- embedding_string = f"Название: {movie['name']}\nГод: {movie['year']}\nЖанры: {movie['genresList']}\nОписание: {movie['description']}"
220
  string_crc32 = calculate_crc32(embedding_string)
221
 
222
  # Проверяем существующий эмбеддинг
@@ -308,7 +308,7 @@ def rerank_results(query, results, conn):
308
  for movie_id, _ in results:
309
  movie = next((m for m in movies_data if m['id'] == movie_id), None)
310
  if movie:
311
- movie_info = f"Название: {movie['name']}\nГод: {movie['year']}\nЖанры: {movie['genresList']}\nОписание: {movie['description']}"
312
  pairs.append([query, movie_info])
313
  movie_ids.append(movie_id)
314
 
@@ -382,7 +382,7 @@ def search_movies(query, top_k=20):
382
  movie = next((m for m in movies_data if m['id'] == movie_id), None)
383
  if movie:
384
  output += f"<h3>{movie['name']} ({movie['year']})</h3>\n"
385
- output += f"<p><strong>Жанры:</strong> {movie['genresList']}</p>\n"
386
  output += f"<p><strong>Описание:</strong> {movie['description']}</p>\n"
387
  output += f"<p><strong>Релевантность:</strong> {score:.4f}</p>\n" # Используем score от реранкера
388
  output += "<hr>\n"
 
216
 
217
  try:
218
  for movie in batch:
219
+ embedding_string = f"Название: {movie['name']}\nГод: {movie['year']}\nЖанры: {movie['genreslist']}\nОписание: {movie['description']}"
220
  string_crc32 = calculate_crc32(embedding_string)
221
 
222
  # Проверяем существующий эмбеддинг
 
308
  for movie_id, _ in results:
309
  movie = next((m for m in movies_data if m['id'] == movie_id), None)
310
  if movie:
311
+ movie_info = f"Название: {movie['name']}\nГод: {movie['year']}\nЖанры: {movie['genreslist']}\nОписание: {movie['description']}"
312
  pairs.append([query, movie_info])
313
  movie_ids.append(movie_id)
314
 
 
382
  movie = next((m for m in movies_data if m['id'] == movie_id), None)
383
  if movie:
384
  output += f"<h3>{movie['name']} ({movie['year']})</h3>\n"
385
+ output += f"<p><strong>Жанры:</strong> {movie['genreslist']}</p>\n"
386
  output += f"<p><strong>Описание:</strong> {movie['description']}</p>\n"
387
  output += f"<p><strong>Релевантность:</strong> {score:.4f}</p>\n" # Используем score от реранкера
388
  output += "<hr>\n"