Spaces:

unik-style
/

unik-ml

Runtime error

RoniFinTech commited on Aug 3, 2023

Commit

44e756c

•

1 Parent(s): d4d34f3

gpu

Files changed (1) hide show

main.py CHANGED Viewed

@@ -63,4 +63,6 @@ def generate(text: str):
     buffer = BytesIO()
     final_image.save(buffer, format="PNG")
-    return StreamingResponse(buffer, 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")