include complete generated text

handler.py

@@ -95,7 +95,8 @@ class EndpointHandler:
 
         answer = generated_text.split('>>END<<')[0].split('>>ANSWER<<:')[1].strip()
 
-        prediction = {'generated_text': answer.replace('<|endoftext|>', ''), 'time': f"{(end-start):.2f} s"} 
+        #prediction = {'generated_text': answer.replace('<|endoftext|>', ''), 'time': f"{(end-start):.2f} s"} 
+        prediction = {'generated_text': answer.replace('<|endoftext|>', ''), 'time': f"{(end-start):.2f} s", 'complete_ouput': generated_text}
 
         result = []
         result.append(prediction)