Update main.py

main.py

@@ -1,7 +1,7 @@
 from ctransformers import AutoModelForCausalLM
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
+from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
-import asyncio
 
 # Carregar o modelo
 llm = AutoModelForCausalLM.from_pretrained(
@@ -40,4 +40,4 @@ Caso não encontre alguma informação, retorne o campo com Null."""
 @app.post("/llm_on_cpu")
 async def stream(item: Validation):
     print("Start")
-    return generate_response(item.prompt)
+    return StreamingResponse(generate_response(item.prompt), media_type="text/plain")