Update app.py

app.py

@@ -59,20 +59,17 @@ 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
+          return highlight
       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'))
-      li.append([word,'    In-claim' if prediction==1 else 'Out-of-Claim', prediction_probability])
+     # li.append([word,'    In-claim' if prediction==1 else 'Out-of-Claim', prediction_probability])
     else:
       highlight.append((word, '    '))
-  headers = ['numeral', 'prediction', 'probability']
-  dff = pd.DataFrame(li)
-  dff.columns = headers
-  return highlight, dff
+  #headers = ['numeral', 'prediction', 'probability']
+  #dff = pd.DataFrame(li)
+ # dff.columns = headers
+  return highlight
 
 
 ##Summarization
@@ -143,7 +140,7 @@ def quad(query,file):
     #stext = resp[0]['summary_text']
     
    # highlight,dff=score_fincat(answer)
-    return answer,summarize_text(answer),get_sustainability(answer),fls(answer)    
+    return answer,summarize_text(answer),score_fincat(answer),get_sustainability(answer),fls(answer)    
                 
                    
 # b6 = gr.Button("Get Sustainability")
@@ -153,7 +150,7 @@ def quad(query,file):
 #iface = gr.Interface(fn=get_sustainability, inputs="textbox", title="CONBERT",description="SUSTAINABILITY TOOL", outputs=gr.HighlightedText(), allow_flagging="never")
 #iface.launch()
 
-iface = gr.Interface(fn=quad, inputs=[gr.inputs.Textbox(label='SEARCH QUERY'),gr.inputs.File(label='TXT FILE')], title="CONBERT",description="SUSTAINABILITY TOOL",article='Article', outputs=[gr.outputs.Textbox(label='Answer'),gr.outputs.Textbox(label='Summary'),gr.HighlightedText(label='SUSTAINABILITY'),gr.HighlightedText(label='FLS')], allow_flagging="never")
+iface = gr.Interface(fn=quad, inputs=[gr.inputs.Textbox(label='SEARCH QUERY'),gr.inputs.File(label='TXT FILE')], title="CONBERT",description="SUSTAINABILITY TOOL",article='Article', outputs=[gr.outputs.Textbox(label='Answer'),gr.outputs.Textbox(label='Summary'),gr.outputs.Textbox(label='NER'),gr.HighlightedText(label='SUSTAINABILITY'),gr.HighlightedText(label='FLS')], allow_flagging="never")
 
 
 iface.launch()