gpu

main.py

@@ -70,14 +70,14 @@ async def generate(text: str):
         image=image,
     ).images[0]
 
-    buffer = BytesIO()
-    final_image.save(buffer, format="PNG")
-    image_bytes = buffer.getvalue()
-
-    return StreamingResponse(BytesIO(image_bytes), media_type="image/png")
+    # buffer = BytesIO()
+    # final_image.save(buffer, format="PNG")
+    # 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")
+    memory_stream = BytesIO()
+    final_image.save(memory_stream, format="PNG")
+    memory_stream.seek(0)
+    return StreamingResponse(memory_stream, media_type="image/png")