Update hideAndSeek.py

hideAndSeek.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 import openai
 from openai import OpenAI
 
-def hide_encrypt(original_input, hide_model, tokenizer):
+def hide(original_input, hide_model, tokenizer):
     hide_template = """<s>Paraphrase the text:%s\n\n"""
     input_text = hide_template % original_input
     inputs = tokenizer(input_text, return_tensors='pt').to(hide_model.device)
@@ -21,8 +21,8 @@ def hide_encrypt(original_input, hide_model, tokenizer):
     hide_input = tokenizer.decode(pred, skip_special_tokens=True)
     return hide_input
 
-def seek_decrypt(hide_input, hide_output, original_input, seek_model, tokenizer):
-    seek_template = """Convert the text:\n%s\n\n%s\n\nConvert the text:\n%s\n\n"""
+def seek(hide_input, hide_output, original_input, seek_model, tokenizer):
+    seek_template = """<s>Convert the text:\n%s\n\n%s\n\nConvert the text:\n%s\n\n"""
     input_text = seek_template % (hide_input, hide_output, original_input)
     inputs = tokenizer(input_text, return_tensors='pt').to(seek_model.device)
     pred = seek_model.generate(