Commit
·
ff6d146
1
Parent(s):
9398f38
Update handler.py
Browse files- 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
|