Update app.py

app.py

@@ -7,9 +7,9 @@ from transformers import BertTokenizer
 
 # Загрузка модели и токенизатора с обработкой ошибки CUDA
 try:
-    model = torch.load('bert_model.pkl')
+    model = torch.load('bert_model.pkl', map_location=torch.device('cpu'))
 except RuntimeError as e:
-    st.error("Ошибка загрузки модели: {e}")
+    st.error(f"Ошибка загрузки модели: {e}")
     model = None
 
 # Загрузка токенизатора BERT
@@ -18,7 +18,7 @@ tokenizer = joblib.load('bert_tokenizer.pkl')
 # Загрузка данных для поиска сходства
 try:
     data = pd.read_excel('DATA_new.xlsx')
-    data_texts = data['Course Title'].tolist()
+    data_texts = data['Text'].tolist()
 except FileNotFoundError:
     st.error("Файл 'DATA_new.xlsx' не найден.")
 except Exception as e: