Update main.py

main.py

@@ -50,6 +50,6 @@ def generate_text(request: Request, prompt: str = Body()):
         output += response.token.text
         yield output
         
-    return output[-2]
+    return output[len(output) - 1]
 
     # return stream