Update app.py

app.py

@@ -23,14 +23,15 @@ app = FastAPI()
 #   FRONTEND_URL = https://your-app.vercel.app
 # For local dev it defaults to localhost:5173
 
-# FRONTEND_URL = os.environ.get("FRONTEND_URL", "https://image-stylizer-deploy.vercel.app")
-FRONTEND_URL = "https://image-stylizer-deploy.vercel.app"
+FRONTEND_URL = os.environ.get("FRONTEND_URL")
+# FRONTEND_URL = "https://image-stylizer-deploy.vercel.app"
 
 app.add_middleware(
     CORSMiddleware,
     allow_origins=[
-        "https://image-stylizer-deploy.vercel.app",
-        "http://localhost:5173",  # for local testing
+        FRONTEND_URL
+        # "https://image-stylizer-deploy.vercel.app",
+        # "http://localhost:5173",  # for local testing
     ],
     allow_credentials=True,
     allow_methods=["*"],
@@ -109,7 +110,7 @@ def save_image_tensor(tensor, path: Path):
     img = tensor.detach().float().cpu()[0].clamp(0, 1).permute(1, 2, 0).numpy() * 255
     Image.fromarray(img.astype("uint8")).save(path)
 
-def stylize_image(img: Image.Image, model, img_size: int = 512):
+def stylize_image(img: Image.Image, model, img_size: int = 256):
     transform = transforms.Compose([
         transforms.Resize(img_size),
         transforms.ToTensor()
@@ -117,7 +118,7 @@ def stylize_image(img: Image.Image, model, img_size: int = 512):
     x = transform(img).unsqueeze(0).to(device)
     with torch.no_grad():
         # autocast only when GPU is available, safe no-op on CPU
-        with torch.cuda.amp.autocast(enabled=use_amp):
+        # with torch.cuda.amp.autocast(enabled=use_amp):
             y = model(x)
     return y