gpu

main.py

@@ -44,7 +44,7 @@ async def root():
 
 
 @app.get("/generate")
-def generate(text: str):
+async def generate(text: str):
     """
     generate image
     """
@@ -75,3 +75,9 @@ def generate(text: str):
     image_bytes = buffer.getvalue()
 
     return StreamingResponse(BytesIO(image_bytes), media_type="image/png")
+    #
+    # memory_stream = BytesIO()
+    # final_image.save(memory_stream, format="PNG")
+    # memory_stream.seek(0)
+    # return StreamingResponse(memory_stream, media_type="image/png")
+