tungdop2 commited on
Commit
486a2da
1 Parent(s): 9025b6e

fix docker

Browse files
Files changed (1) hide show
  1. model.py +5 -0
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