Skip to content

yuanjie-ai/torch4uie

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

image image image

🔥UIE🔥

Install

pip install -U torch4uie

Usages

from uie import UIE

schema = ['时间', '选手', '赛事名称']
ie = UIE(model='uie-mini', schema=schema, device='cpu')
ie("2月8日上午北京冬奥会自由式滑雪女子大跳台决赛中中国选手谷爱凌以188.25分获得金牌!")
# [{'时间': [{'text': '2月8日上午', 'start': 0, 'end': 6, 'probability': 0.986075}],
#   '选手': [{'text': '谷爱凌', 'start': 28, 'end': 31, 'probability': 0.994467}],
#   '赛事名称': [{'text': '北京冬奥会自由式滑雪女子大跳台决赛',
#     'start': 6,
#     'end': 23,
#     'probability': 0.58514273}]}]

Releases

No releases published

Packages

No packages published