ekaterina-simonova commited on
Commit
08f45ea
·
verified ·
1 Parent(s): 75cfe15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -47,8 +47,8 @@ try:
47
  "cointegrated/LaBSE-en-ru",
48
  cache_folder="/tmp/hf_cache_force"
49
  )
50
- # Сначала загружаем модель, потом перемещаем на устройство
51
- test_model = test_model.to(device)
52
 
53
  # Проверяем работоспособность
54
  test_text = ["тестовый текст"]
@@ -505,12 +505,11 @@ def load_models():
505
  # Добавляем определение start_time
506
  start_time = time.time()
507
 
508
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
509
  model = SentenceTransformer(
510
  EMBEDDING_MODEL,
511
  cache_folder=os.path.expanduser("~/.cache/huggingface/hub")
512
  )
513
- model = model.to(device)
514
 
515
  # Проверяем работоспособность
516
  test_text = ["тестовый текст"]
@@ -519,6 +518,7 @@ def load_models():
519
 
520
  logger.info(f"Модель загружена за {time.time()-start_time:.2f} сек")
521
  logger.info(f"Размерность эмбеддингов: {model.get_sentence_embedding_dimension()}")
 
522
 
523
  # 2. Загрузка FAISS индекса
524
  logger.info(f"Загрузка FAISS индекса: {FAISS_INDEX_PATH}")
 
47
  "cointegrated/LaBSE-en-ru",
48
  cache_folder="/tmp/hf_cache_force"
49
  )
50
+ # Изменяем порядок инициализации
51
+ test_model = test_model.to('cpu') # Сначала явно переносим на CPU
52
 
53
  # Проверяем работоспособность
54
  test_text = ["тестовый текст"]
 
505
  # Добавляем определение start_time
506
  start_time = time.time()
507
 
 
508
  model = SentenceTransformer(
509
  EMBEDDING_MODEL,
510
  cache_folder=os.path.expanduser("~/.cache/huggingface/hub")
511
  )
512
+ model = model.to('cpu') # Сначала явно переносим на CPU
513
 
514
  # Проверяем работоспособность
515
  test_text = ["тестовый текст"]
 
518
 
519
  logger.info(f"Модель загружена за {time.time()-start_time:.2f} сек")
520
  logger.info(f"Размерность эмбеддингов: {model.get_sentence_embedding_dimension()}")
521
+
522
 
523
  # 2. Загрузка FAISS индекса
524
  logger.info(f"Загрузка FAISS индекса: {FAISS_INDEX_PATH}")