rajistics commited on
Commit
79c2061
·
1 Parent(s): 88a4d3b

changed spans

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,7 +37,7 @@ def fin_ner(text):
37
  ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
38
  #api = gr.Interface.load("dslim/bert-base-NER", src='models')
39
  spans = ner_pipeline(text)
40
- print ("spans")
41
  #replaced_spans = [(key, None) if value=='No Disease' else (key, value) for (key, value) in spans]
42
  return spans
43
 
@@ -75,7 +75,7 @@ with demo:
75
 
76
  b4 = gr.Button("Extract Companies & Segments")
77
  replaced_spans = gr.HighlightedText()
78
- b4.click(fin_ner, inputs=text, outputs=replaced_spans)
79
 
80
  b5 = gr.Button("Extract Financial Sentiment")
81
  fin_spans = gr.HighlightedText()
 
37
  ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
38
  #api = gr.Interface.load("dslim/bert-base-NER", src='models')
39
  spans = ner_pipeline(text)
40
+ print (spans)
41
  #replaced_spans = [(key, None) if value=='No Disease' else (key, value) for (key, value) in spans]
42
  return spans
43
 
 
75
 
76
  b4 = gr.Button("Extract Companies & Segments")
77
  replaced_spans = gr.HighlightedText()
78
+ b4.click(fin_ner, inputs=text, outputs=spans)
79
 
80
  b5 = gr.Button("Extract Financial Sentiment")
81
  fin_spans = gr.HighlightedText()