Update fincat_utils.py

fincat_utils.py

@@ -99,10 +99,12 @@ def bert_embedding_extract(context_text, word):
   list_token_embeddings = get_bert_embeddings(tokens_tensor, segments_tensors, model)
   word_tokens,tt,st = bert_text_preparation(word, tokenizer)
   word_embedding_all = []
-  for word_tk in word_tokens:
-    word_index = tokenized_text.index(word_tk)
-    word_embedding = list_token_embeddings[word_index]
-    word_embedding_all.append(word_embedding)
-  word_embedding_mean = np.array(word_embedding_all).mean(axis=0)
-  return word_embedding_mean
-
+  try:
+      for word_tk in word_tokens:
+        word_index = tokenized_text.index(word_tk)
+        word_embedding = list_token_embeddings[word_index]
+        word_embedding_all.append(word_embedding)
+      word_embedding_mean = np.array(word_embedding_all).mean(axis=0)
+      return word_embedding_mean
+  except:
+      return ['None']