debug

app.py

@@ -37,13 +37,16 @@ If a question does not make any sense, or is not factually coherent, explain why
 """
 LEN_DEFAULT_PROMPT = len(tokenizer.encode(DEFAULT_SYSTEM_PROMPT))
 def embed(user, max_length, window_size, method, prompt):
-    print("prompt:", prompt)
+    
     uid = USERS.index(user)
     watermarker = Watermarker(tokenizer=tokenizer, model=model, window_size=window_size, payload_bits=PAYLOAD_BITS)
     prompt = get_prompt(prompt)
+    print("prompt:", prompt)
     watermarked_texts = watermarker.embed(key=args.key, messages=[ uid ],
                                           max_length=max_length+LEN_DEFAULT_PROMPT, method=method, prompt=prompt)
     print("===")
+    print(watermarked_text)
+    print("===")
     print(watermarked_texts[0].split("[/INST]")[1][0])
     print("===")
     print(watermarked_texts[0].split("[/INST]"))