Spaces:

Shredder
/

CONBERT

Runtime error

Shredder commited on Sep 5, 2022

Commit

79c9105

1 Parent(s): 6b3f117

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -59,7 +59,7 @@ def score_fincat(txt):
       features = bert_embedding_extract(context_text, word)
       if(features[0]=='None'):
           highlight.append(('None', '    '))
-          dff = pd.DataFrame([['None', 'None', 'None'])
           headers = ['numeral', 'prediction', 'probability']
           dff.columns = headers
           return highlight, dff

       features = bert_embedding_extract(context_text, word)
       if(features[0]=='None'):
           highlight.append(('None', '    '))
+          dff = pd.DataFrame([['None', 'None', 'None']])
           headers = ['numeral', 'prediction', 'probability']
           dff.columns = headers
           return highlight, dff