Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ def check_plagiarism(text):
|
|
| 51 |
|
| 52 |
|
| 53 |
# Extract the generated tokens from the model's predictions
|
| 54 |
-
generated_tokens = outputs
|
| 55 |
|
| 56 |
# Compute masks for watermark detection
|
| 57 |
eos_token_mask = logits_processor.compute_eos_token_mask(
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
# Extract the generated tokens from the model's predictions
|
| 54 |
+
generated_tokens = outputs[:, inputs_len:]
|
| 55 |
|
| 56 |
# Compute masks for watermark detection
|
| 57 |
eos_token_mask = logits_processor.compute_eos_token_mask(
|