return prediction in an array

handler.py

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