Update main.py

main.py

@@ -1,4 +1,5 @@
-from fastapi import FastAPI, Request, Response, Form, File, UploadFile
+from fastapi import FastAPI, Request, Form, File, UploadFile
+from fastapi.responses import StreamingResponse
 from contextlib import asynccontextmanager
 from starlette.middleware.cors import CORSMiddleware
 
@@ -58,8 +59,8 @@ async def text_to_image(
 
     bytes = BytesIO()
     image.save(bytes, "PNG")
-    # bytes.seek(0)
-    return Response(content=bytes, media_type="image/png")
+    bytes.seek(0)
+    return StreamingResponse(bytes, media_type="image/png")
 
 
 @app.post("/image-to-image/")
@@ -84,8 +85,8 @@ async def image_to_image(
 
     bytes = BytesIO()
     image.save(bytes, "PNG")
-    # bytes.seek(0)
-    return Response(content=bytes, media_type="image/png")
+    bytes.seek(0)
+    return StreamingResponse(bytes, media_type="image/png")
 
 
 @app.post("/inpainting/")
@@ -115,5 +116,5 @@ async def inpainting(
 
     bytes = BytesIO()
     image.save(bytes, "PNG")
-    # bytes.seek(0)
-    return Response(content=bytes, media_type="image/png")
+    bytes.seek(0)
+    return StreamingResponse(bytes, media_type="image/png")