Update main.py

main.py

@@ -59,7 +59,7 @@ async def text_to_image(
     bytes = BytesIO()
     image.save(bytes, "PNG")
     # bytes.seek(0)
-    return Response(bytes, media_type="image/png")
+    return Response(content=bytes, media_type="image/png")
 
 
 @app.post("/image-to-image/")
@@ -85,7 +85,7 @@ async def image_to_image(
     bytes = BytesIO()
     image.save(bytes, "PNG")
     # bytes.seek(0)
-    return Response(bytes, media_type="image/png")
+    return Response(content=bytes, media_type="image/png")
 
 
 @app.post("/inpainting/")
@@ -116,4 +116,4 @@ async def inpainting(
     bytes = BytesIO()
     image.save(bytes, "PNG")
     # bytes.seek(0)
-    return Response(bytes, media_type="image/png")
+    return Response(content=bytes, media_type="image/png")