Update api.py

api.py

@@ -293,14 +293,11 @@ async def async_generator_wrapper(prompt: str):
         yield text_piece
         await asyncio.sleep(0.1)
 
-@app.get("/generate")
+
+from fastapi.responses import PlainTextResponse
+
+@app.get("/generate", response_class=PlainTextResponse)
 async def generate(request: Request):
     prompt = request.query_params.get("prompt", "안녕하세요")
     response_text = respond(prompt)
-    
-    async def stream_response():
-        for chunk in respond(response_text):
-            yield chunk
-            await asyncio.sleep(0.1)
-
-    return StreamingResponse(stream_response(), media_type="text/plain")
+    return response_text