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