adding print
Browse files
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 |
-
|
37 |
#api = gr.Interface.load("dslim/bert-base-NER", src='models')
|
38 |
-
spans =
|
|
|
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 |
|