Update main.py

main.py

@@ -66,5 +66,6 @@ def generate_text(
     # repetition_penalty = body.get("repetition_penalty", 1.0)
     history = []  # You might need to handle this based on your actual usage
     generatedOutput = generater(message, history, temperature, top_p, top_k)
+    actualText = generatedOutput[len(generatedOutput) - 1]
 
-    return {"generated_text": generatedOutput}
+    return {"generated_text": actualText}