Update main.py

main.py

@@ -65,4 +65,4 @@ def generate_text(
     history = []  # You might need to handle this based on your actual usage
     generatedOutput = generater(message, history, temperature, top_p, top_k)
 
-    return {"generated_text": generatedOutput[-1]}
+    return {"generated_text": generatedOutput}