gpu

main.py

@@ -3,6 +3,7 @@ from io import BytesIO
 import torch
 from diffusers import DiffusionPipeline
 from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 
 # load both base & refiner
@@ -28,6 +29,14 @@ refiner.enable_attention_slicing()
 # Create a new FastAPI app instance
 app = FastAPI()
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+    allow_credentials=True,
+)
+
 
 @app.get("/")
 async def root():