Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,7 @@ Cristiano Ronaldo dos Santos Aveiro (Portuguese pronunciation: [kɾiʃˈtjɐnu
|
|
43 |
|
44 |
labels = ["person", "award", "date", "competitions", "teams"]
|
45 |
|
46 |
-
entities = model.predict_entities(text, labels, threshold=0.
|
47 |
|
48 |
for entity in entities:
|
49 |
print(entity["text"], "=>", entity["label"])
|
|
|
43 |
|
44 |
labels = ["person", "award", "date", "competitions", "teams"]
|
45 |
|
46 |
+
entities = model.predict_entities(text, labels, threshold=0.3)
|
47 |
|
48 |
for entity in entities:
|
49 |
print(entity["text"], "=>", entity["label"])
|