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

adding print

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -31,11 +31,13 @@ def text_to_sentiment(text):
31
 
32
  ##Company Extraction
33
  def fin_ner(text):
 
34
  tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
35
  model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
36
- ner = pipeline("ner", model=model, tokenizer=tokenizer)
37
  #api = gr.Interface.load("dslim/bert-base-NER", src='models')
38
- spans = ner(text)
 
39
  #replaced_spans = [(key, None) if value=='No Disease' else (key, value) for (key, value) in spans]
40
  return spans
41
 
 
31
 
32
  ##Company Extraction
33
  def fin_ner(text):
34
+ print ("ner")
35
  tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
36
  model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
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