Ihor commited on
Commit
0a058f4
1 Parent(s): e7fd176

Update README.md

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