Update main.py

main.py

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