Update handler.py
Browse files- handler.py +3 -1
handler.py
CHANGED
@@ -130,6 +130,8 @@ class EndpointHandler:
|
|
130 |
|
131 |
# Extract the relevant part of the assistant's answer
|
132 |
assistant_answer_start = generated_text[0].find("ASSISTANT:") + len("ASSISTANT:")
|
133 |
-
assistant_answer = generated_text[0][assistant_answer_start:].strip()
|
|
|
|
|
134 |
|
135 |
return [{"generated_text": assistant_answer}]
|
|
|
130 |
|
131 |
# Extract the relevant part of the assistant's answer
|
132 |
assistant_answer_start = generated_text[0].find("ASSISTANT:") + len("ASSISTANT:")
|
133 |
+
assistant_answer = generated_text[0][assistant_answer_start:].strip()
|
134 |
+
|
135 |
+
print("model answer", assistant_answer)
|
136 |
|
137 |
return [{"generated_text": assistant_answer}]
|