shibing624 commited on
Commit
8db4031
1 Parent(s): 393188a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ sys.path.append('..')
45
  from textgen.language_modeling import SongNetModel
46
 
47
 
48
- model = SongNetModel(model_type='songnet', model_name='songnet-base-chinese-couplet')
49
  sentences = [
50
  "严蕊<s1>如梦令<s2>道是梨花不是。</s>道是杏花不是。</s>白白与红红,别是东风情味。</s>曾记。</s>曾记。</s>人在武陵微醉。",
51
  "<s1><s2>一句相思吟岁月</s>千杯美酒醉风情",
 
45
  from textgen.language_modeling import SongNetModel
46
 
47
 
48
+ model = SongNetModel(model_type='songnet', model_name='shibing624/songnet-base-chinese-couplet')
49
  sentences = [
50
  "严蕊<s1>如梦令<s2>道是梨花不是。</s>道是杏花不是。</s>白白与红红,别是东风情味。</s>曾记。</s>曾记。</s>人在武陵微醉。",
51
  "<s1><s2>一句相思吟岁月</s>千杯美酒醉风情",