Spaces:

unik-style
/

unik-ml

Runtime error

RoniFinTech commited on Aug 3, 2023

Commit

d4d34f3

•

1 Parent(s): e15aeab

gpu

Files changed (1) hide show

main.py CHANGED Viewed

@@ -61,6 +61,6 @@ def generate(text: str):
         image=image,
     ).images[0]
-    return StreamingResponse(BytesIO(final_image), media_type="image/png")
-    # Return the generated text in a JSON response
-    # return {"output": output[0]["generated_text"]}

         image=image,
     ).images[0]
+    buffer = BytesIO()
+    final_image.save(buffer, format="PNG")
+    return StreamingResponse(buffer, media_type="image/png")