EnariGmbH commited on
Commit
69d5204
·
verified ·
1 Parent(s): 2554896

Update handler.py

Browse files
Files changed (1) hide show
  1. 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}]