ArneBinder
commited on
Commit
•
fa59c20
1
Parent(s):
ca1427e
adjust for recent version of pytorch-ie
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ ner_pipeline = AutoPipeline.from_pretrained(model_name_or_path, device=-1, num_w
|
|
22 |
def predict(text):
|
23 |
document = ExampleDocument(text)
|
24 |
|
25 |
-
ner_pipeline(document
|
26 |
|
27 |
doc = {
|
28 |
"text": document.text,
|
|
|
22 |
def predict(text):
|
23 |
document = ExampleDocument(text)
|
24 |
|
25 |
+
ner_pipeline(document)
|
26 |
|
27 |
doc = {
|
28 |
"text": document.text,
|