Ihor commited on
Commit
e423c87
1 Parent(s): ed920d8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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.25)
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"])