xusenlin commited on
Commit
01e3caf
1 Parent(s): c054bec

Update README.md

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