Spaces:
Build error
Build error
fix docker
Browse files
model.py
CHANGED
|
@@ -57,4 +57,9 @@ class ChallengePromptGenerator:
|
|
| 57 |
|
| 58 |
# Decode
|
| 59 |
decoded_outputs = self.tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 60 |
return decoded_outputs
|
|
|
|
| 57 |
|
| 58 |
# Decode
|
| 59 |
decoded_outputs = self.tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
| 60 |
+
outputs = []
|
| 61 |
+
for out in decoded_outputs:
|
| 62 |
+
if out[-1] != ".":
|
| 63 |
+
out = ".".join(out.split(".")[:-1]) + "."
|
| 64 |
+
outputs.append(out)
|
| 65 |
return decoded_outputs
|