Update main.py

main.py

@@ -66,6 +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]
+    actualText = list(generatedOutput)[-1]
 
     return {"generated_text": actualText}