Update app.py

app.py

@@ -15,6 +15,7 @@ from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import StreamingResponse
+from fastapi.responses import PlainTextResponse
 
 import uvicorn 
 
@@ -260,7 +261,8 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
                 buffer += new_text
                 yield buffer.encode('utf-8')
 
-    return StreamingResponse(generate(), media_type="text/plain")
+    #return StreamingResponse(generate(), media_type="text/plain")
+    return PlainTextResponse(generated_text)
 
 @app.post("/chat/")
 async def test_endpoint(message: dict):