Update app.py

app.py

@@ -57,6 +57,10 @@ def score_fincat(txt):
       x = {'paragraph' : txt, 'offset_start':st, 'offset_end':ed}
       context_text = extract_context_words(x)
       features = bert_embedding_extract(context_text, word)
+      if(features[0]=='None'):
+          highlight.append(('None', '    '))
+          dff = pd.DataFrame(['None'])
+          return highlight, dff
       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'))