tingxinli commited on
Commit
2dd1672
1 Parent(s): e4f6d6a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md CHANGED
@@ -45,6 +45,26 @@ print(hide_input)
45
  # '李明用华为(Mate 40)换了一箱好吃的橙子。
46
  ```
47
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
48
  下面是一个完整调用Hide-and-Seek框架的例子。注意完整的隐私保护流程demo需要自备OpenAI的API token。
49
  ```ipython
50
  # see hideAndSeek.py in this repo
 
45
  # '李明用华为(Mate 40)换了一箱好吃的橙子。
46
  ```
47
 
48
+ 下面是单独调用seek模型完成摘要解密的一个例子。
49
+ ```ipython
50
+ from transformers import AutoTokenizer, AutoModelForCausalLM
51
+ tokenizer = AutoTokenizer.from_pretrained("tingxinli/seek-820m")
52
+ model = AutoModelForCausalLM.from_pretrained("tingxinli/seek-820m").to('cuda:0')
53
+ seek_template = "Convert the text:\n%s\n\n%s\n\nConvert the text:\n%s\n\n"
54
+ hide_input = "前天,'2022北京海淀·颐和园经贸合作洽谈会成功举行,各大媒体竞相报道了活动盛况,小李第一时间将昨天媒体报道情况进行了整理。人民日报 中国青年网 国际联合报 北京商报 消费者观察报 上海晚报 杭州日报 海峡晚报 北京日报 北京市电视一台?北京新闻 人民网 手机雅虎网 网易北京 长三角经济网 新京网 中国农业新闻网 北京圆桌 居然有这么多!还有部分媒体将在未来一周陆续发稿,为经洽会点!为海淀点!阅读投诉阅读精选留言加载中以上留言由公众号筛选后显示了解留言功能详情"
55
+ hide_output = "2022北京海淀·颐和园经贸合作洽谈会成功举办,各大媒体广泛报道"
56
+ original_input = "昨天,’2016苏州吴中·太湖经贸合作洽谈会成功举行,各大媒体竞相报道了活动盛况,小吴第一时间将今天媒体报道情况进行了整理。新华社 中国青年报?中青在线 香港大公报?大公网 香港商报 消费者导报 扬子晚报 江南时报 苏州日报 姑苏晚报 城市商报 苏州广电一套?苏州新闻 新华网 手机凤凰网 网易苏州 长三角城市网 新苏网 中国商务新闻网 苏州圆桌 居然有这么多!还有部分媒体将在今后几天陆续发稿,为经洽会点!为吴中点!阅读投诉阅读精选留言加载中以上留言由公众号筛选后显示了解留言功能详情"
57
+ input_text = seek_template % (hide_input, hide_output, original_input)
58
+ inputs = tokenizer(input_text, return_tensors='pt').to('cuda:0')
59
+ pred = model.generate(**inputs, max_length=512)
60
+ pred = pred.cpu()[0][len(inputs['input_ids'][0]):]
61
+ original_output = tokenizer.decode(pred, skip_special_tokens=True)
62
+ print(original_output)
63
+
64
+ # output:
65
+ # 2016苏州吴中·太湖经贸合作洽谈会成功举办,各大媒体广泛报道
66
+ ```
67
+
68
  下面是一个完整调用Hide-and-Seek框架的例子。注意完整的隐私保护流程demo需要自备OpenAI的API token。
69
  ```ipython
70
  # see hideAndSeek.py in this repo