Antoine Chaffin commited on
Commit
7e6b28b
1 Parent(s): 1af9b4f

Remove window size from embed call

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def embed(user, max_length, window_size, method, prompt):
38
  watermarker = Watermarker(tokenizer=tokenizer, model=model, window_size=window_size, payload_bits=PAYLOAD_BITS)
39
  prompt = get_prompt(prompt)
40
  watermarked_texts = watermarker.embed(key=args.key, messages=[ uid ],
41
- max_length=max_length, method=method, prompt=prompt, window_size=window_size)
42
  print("watermarked_texts: ", watermarked_texts)
43
 
44
  return watermarked_texts[0]
 
38
  watermarker = Watermarker(tokenizer=tokenizer, model=model, window_size=window_size, payload_bits=PAYLOAD_BITS)
39
  prompt = get_prompt(prompt)
40
  watermarked_texts = watermarker.embed(key=args.key, messages=[ uid ],
41
+ max_length=max_length, method=method, prompt=prompt)
42
  print("watermarked_texts: ", watermarked_texts)
43
 
44
  return watermarked_texts[0]