Update README.md

README.md

@@ -93,12 +93,12 @@ HaS-820m").to('cuda:0')
 
 original_input = "华纳兄弟影业（Warner Bro）著名的作品有《蝙蝠侠》系列、《超人》系列、《黑客帝国》系列和《指环王》系列。目前华纳未考虑推出《蝙蝠侠》系列新作。"
 print('original input:', original_input)
-hide_input = hide_encrypt(original_input, model, tokenizer)
+hide_input = hide(original_input, model, tokenizer)
 print('hide input:', hide_input)
 prompt = "Translate the following text into English.\n %s\n" % hide_input
 hide_output = get_gpt_output(prompt)
 print('hide output:', hide_output)
-original_output = seek_decrypt(hide_input, hide_output, original_input, model, tokenizer)
+original_output = seek(hide_input, hide_output, original_input, model, tokenizer)
 print('original output:', original_output)
 
 # output: