Update score_fincat.py

score_fincat.py

@@ -23,11 +23,12 @@ def score_fincat(txt):
       context_text = extract_context_words(x)
       features = bert_embedding_extract(context_text, word)
       if(features[0]=='None'):
-          highlight.append((txt, '    '))
-          return highlight
+          continue
       prediction = lr_clf.predict(features.reshape(1, 768))
       prediction_probability = '{:.4f}'.format(round(lr_clf.predict_proba(features.reshape(1, 768))[:,1][0], 4))
       highlight.append((word, '    In-claim' if prediction==1 else 'Out-of-Claim'))
     else:
-      highlight.append((word, '    '))
+      continue
+  if(len(highlight)<1):
+      highlight.append((txt,'None'))    
   return highlight