Fixed the issue where `tt_labels` was mistakenly written as `labels`
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ text = """
|
|
| 39 |
|
| 40 |
tta_labels = ["ARTIFACTS", "ANIMAL", "CIVILIZATION", "DATE", "EVENT", "STUDY_FIELD", "LOCATION", "MATERIAL", "ORGANIZATION", "PERSON", "PLANT", "QUANTITY", "TIME", "TERM", "THEORY"]
|
| 41 |
|
| 42 |
-
entities = model.predict_entities(text,
|
| 43 |
|
| 44 |
for entity in entities:
|
| 45 |
print(entity["text"], "=>", entity["label"])
|
|
|
|
| 39 |
|
| 40 |
tta_labels = ["ARTIFACTS", "ANIMAL", "CIVILIZATION", "DATE", "EVENT", "STUDY_FIELD", "LOCATION", "MATERIAL", "ORGANIZATION", "PERSON", "PLANT", "QUANTITY", "TIME", "TERM", "THEORY"]
|
| 41 |
|
| 42 |
+
entities = model.predict_entities(text, tta_labels)
|
| 43 |
|
| 44 |
for entity in entities:
|
| 45 |
print(entity["text"], "=>", entity["label"])
|