Nelsonlin0321
commited on
Commit
·
0aa787d
1
Parent(s):
3b98af5
Upload handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
@@ -46,3 +46,4 @@ class EndpointHandler():
|
|
46 |
for s in generation_output.sequences:
|
47 |
output = self.tokenizer.decode(s)
|
48 |
output = output.split("### Response:")[1].strip()
|
|
|
|
46 |
for s in generation_output.sequences:
|
47 |
output = self.tokenizer.decode(s)
|
48 |
output = output.split("### Response:")[1].strip()
|
49 |
+
return output
|