Spaces:
Build error
Build error
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 |
|