Spaces:
Runtime error
Runtime error
ArneBinder
commited on
Commit
•
7c8f86e
1
Parent(s):
47a773b
adjust for recent version of pytorch-ie
Browse files
app.py
CHANGED
@@ -27,12 +27,12 @@ re_pipeline = AutoPipeline.from_pretrained(re_model_name_or_path, device=-1, num
|
|
27 |
def predict(text):
|
28 |
document = ExampleDocument(text)
|
29 |
|
30 |
-
ner_pipeline(document
|
31 |
|
32 |
for entity in document.entities.predictions:
|
33 |
document.entities.append(entity)
|
34 |
|
35 |
-
re_pipeline(document
|
36 |
|
37 |
t = PrettyTable()
|
38 |
t.field_names = ["head", "tail", "relation"]
|
|
|
27 |
def predict(text):
|
28 |
document = ExampleDocument(text)
|
29 |
|
30 |
+
ner_pipeline(document)
|
31 |
|
32 |
for entity in document.entities.predictions:
|
33 |
document.entities.append(entity)
|
34 |
|
35 |
+
re_pipeline(document)
|
36 |
|
37 |
t = PrettyTable()
|
38 |
t.field_names = ["head", "tail", "relation"]
|