Adding padding token

app.py

@@ -32,6 +32,7 @@ If a question does not make any sense, or is not factually coherent, explain why
 model = AutoModelForCausalLM.from_pretrained(args.model, use_auth_token=hf_token, torch_dtype=torch.float16,
         device_map='auto').to(device)
 tokenizer = AutoTokenizer.from_pretrained(args.model, use_auth_token=hf_token)
+tokenizer.pad_token = tokenizer.eos_token
 
 def embed(user, max_length, window_size, method, prompt):
     uid = USERS.index(user)
@@ -40,7 +41,8 @@ def embed(user, max_length, window_size, method, prompt):
     watermarked_texts = watermarker.embed(key=args.key, messages=[ uid ],
                                           max_length=max_length, method=method, prompt=prompt)
     print("watermarked_texts: ", watermarked_texts)
-
+    print(watermarked_texts[0].replace(prompt, ""))
+    watermarked_texts[0].split("[/INST]")[-1]
     return watermarked_texts[0].replace(prompt, "")
     
 def detect(attacked_text, window_size, method, prompt):