Update stri.py

stri.py

@@ -29,7 +29,7 @@ for annotation in annot:
     with torch.no_grad():
         outputs = model(**annotation_tokens)
         hidden_states = outputs.hidden_states
-        last_hidden_state = hidden_states[-2]
+        last_hidden_state = hidden_states[-1]
         embeddings.append(torch.mean(last_hidden_state, dim=1).squeeze())
 
 # Получение эмбеддинга запроса от пользователя
@@ -47,7 +47,7 @@ if query:
     with torch.no_grad():
         query_outputs = model(**query_tokens)
         query_hidden_states = query_outputs.hidden_states
-        query_last_hidden_state = query_hidden_states[-2]
+        query_last_hidden_state = query_hidden_states[-1]
         query_embedding = torch.mean(query_last_hidden_state, dim=1).squeeze()
 
     # Вычисление косинусного расстояния между эмбеддингом запроса и каждой аннотацией