Spaces:
Runtime error
Runtime error
Antoine Chaffin
commited on
Commit
•
7e6b28b
1
Parent(s):
1af9b4f
Remove window size from embed call
Browse files
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
|
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]
|