Create README.md

README.md

@@ -0,0 +1,17 @@
+chinese-simile-generative 是一个将句子A改写成带有修辞手法(主要为比喻，明喻)的句子B的seq2seq模型。
+
+A： 我走得很慢，慢极了。
+B： 我走的很慢，像蜗牛一样。
+
+  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+  tokenizer = AutoTokenizer.from_pretrained("figurative-nlp/chinese-simile-generation")
+  model = AutoModelForSeq2SeqLM.from_pretrained("figurative-nlp/chinese-simile-generation")
+  
+  
+  input_ids = tokenizer(
+      "我走得很慢，慢极了", return_tensors="pt"
+  ).input_ids
+  outputs = model.generate(input_ids,num_beams = 5,max_length = 64)
+  result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+  print(result)
+  #result : 我走的很慢，像蜗牛一样。