Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -255,6 +255,7 @@ class EndpointHandler:
|
|
| 255 |
|
| 256 |
# Extract the generated part (after the prompt)
|
| 257 |
response_text = decoded_output[len(prompt):].strip()
|
|
|
|
| 258 |
|
| 259 |
# Parse the response to extract safety assessment
|
| 260 |
safety_result = self.extract_and_parse_json(response_text)
|
|
|
|
| 255 |
|
| 256 |
# Extract the generated part (after the prompt)
|
| 257 |
response_text = decoded_output[len(prompt):].strip()
|
| 258 |
+
print(response_text)
|
| 259 |
|
| 260 |
# Parse the response to extract safety assessment
|
| 261 |
safety_result = self.extract_and_parse_json(response_text)
|