Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
| 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(
|
| 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}")
|