Update README.md
Browse files
README.md
CHANGED
|
@@ -25,8 +25,8 @@ license: apache-2.0
|
|
| 25 |
```python
|
| 26 |
from transformers import AutoModel, AutoTokenizer
|
| 27 |
|
| 28 |
-
tokenizer = AutoTokenizer.from_pretrained("uie-base", trust_remote_code=True)
|
| 29 |
-
model = AutoModel.from_pretrained("uie-base", trust_remote_code=True)
|
| 30 |
|
| 31 |
schema = ["时间", "选手", "赛事名称"] # Define the schema for entity extraction
|
| 32 |
print(model.predict(tokenizer, "2月8日上午北京冬奥会自由式滑雪女子大跳台决赛中中国选手谷爱凌以188.25分获得金牌！", schema=schema))
|
|
|
|
| 25 |
```python
|
| 26 |
from transformers import AutoModel, AutoTokenizer
|
| 27 |
|
| 28 |
+
tokenizer = AutoTokenizer.from_pretrained("xusenlin/uie-base", trust_remote_code=True)
|
| 29 |
+
model = AutoModel.from_pretrained("xusenlin/uie-base", trust_remote_code=True)
|
| 30 |
|
| 31 |
schema = ["时间", "选手", "赛事名称"] # Define the schema for entity extraction
|
| 32 |
print(model.predict(tokenizer, "2月8日上午北京冬奥会自由式滑雪女子大跳台决赛中中国选手谷爱凌以188.25分获得金牌！", schema=schema))
|