auhide commited on
Commit
8285da4
1 Parent(s): 0f670ee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -123,5 +123,7 @@ print("NERs:", predict(text, model=model, tokenizer=tokenizer))
123
  ```
124
  ```sh
125
  Input: Барух Спиноза е роден в Амстердам .
126
- NERs: [{'word': 'Барух Спиноза', 'entity': 'PER'}, {'word': 'Амстердам', 'entity': 'LOC'}]
127
- ```
 
 
 
123
  ```
124
  ```sh
125
  Input: Барух Спиноза е роден в Амстердам .
126
+ NERs: [{'word': 'Барух Спиноза', 'entity_group': 'PER'}, {'word': 'Амстердам', 'entity_group': 'LOC'}]
127
+ ```
128
+
129
+ Note: There are three types of entities - `PER`, `ORG`, `LOC`.