Update app.py

app.py

@@ -12,7 +12,7 @@ import os
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")
 
-imsize = 512
+imsize = 256
 beta = 1e5 # Style weight multiplier
 
 # Define the style layers and their weights
@@ -52,7 +52,7 @@ except Exception as e:
 
 # --- Helper Functions ---
 
-def image_loader(image: Image.Image, size=512, device=torch.device("cpu")):
+def image_loader(image: Image.Image, size=256, device=torch.device("cpu")):
     """Loads a PIL Image, resizes, converts to tensor, and normalizes."""
     # VGG19 mean and std
     normalize = T.Normalize(mean=[0.485, 0.456, 0.406],
@@ -134,7 +134,7 @@ def stylize_image(content_image: Image.Image):
         optimizer = optim.Adam([generated_img], lr=lr)
 
         # 4. Run optimization loop
-        inference_steps = 500 # Number of optimization steps for inference
+        inference_steps = 200 # Number of optimization steps for inference
 
         for step in range(1, inference_steps + 1):
             # Get features for the generated image