svjack commited on
Commit
857f7c5
·
1 Parent(s): e5cb100

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -0
README.md ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - zh
4
+ library_name: transformers
5
+ pipeline_tag: text2text-generation
6
+ ---
7
+
8
+ ```python
9
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
10
+
11
+ tokenizer = AutoTokenizer.from_pretrained("svjack/squad_gen_qst_zh_v0")
12
+
13
+ model = AutoModelForSeq2SeqLM.from_pretrained("svjack/squad_gen_qst_zh_v0")
14
+
15
+ tokenizer.decode(
16
+ model.generate(
17
+ tokenizer.encode(
18
+ "川普是美利坚共和国的总统,并在56岁时获得该奖项。|美利坚",
19
+ return_tensors="pt", add_special_tokens=True
20
+ ))[0],
21
+ skip_special_tokens = True
22
+ )
23
+
24
+ '''
25
+ '川普在哪个国家当总统?'
26
+ '''
27
+ ```