Update README.md

README.md

@@ -3,15 +3,15 @@ chinese-simile-generative 是一个将句子A改写成带有修辞手法(主要
 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 : 我走的很慢，像蜗牛一样。
+      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 : 我走的很慢，像蜗牛一样。