Devops-hestabit commited on
Commit
ff6d146
·
1 Parent(s): 9398f38

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -48,7 +48,7 @@ class EndpointHandler():
48
  pad_token_id = 50256,
49
  num_return_sequences = 1
50
  )
51
- decoded_output = tokenizer.decode(encoded_output[0], skip_special_tokens=True).replace(prompt,"")
52
  decoded_output = decoded_output.split("Alice Gate:", 1)[1].split(f"{user_name}:",1)[0].strip()
53
  parsed_result = re.sub('\*.*?\*', '', decoded_output).strip()
54
  if len(parsed_result) != 0: decoded_output = parsed_result
 
48
  pad_token_id = 50256,
49
  num_return_sequences = 1
50
  )
51
+ decoded_output = self.tokenizer.decode(encoded_output[0], skip_special_tokens=True).replace(prompt,"")
52
  decoded_output = decoded_output.split("Alice Gate:", 1)[1].split(f"{user_name}:",1)[0].strip()
53
  parsed_result = re.sub('\*.*?\*', '', decoded_output).strip()
54
  if len(parsed_result) != 0: decoded_output = parsed_result