Update app.py

app.py

@@ -1655,48 +1655,48 @@ class StyleTransferSystem:
         return model
 
     def apply_adain_style(self, content_image, style_image, model, alpha=1.0):
-    """Apply AdaIN-based style transfer"""
-    if content_image is None or style_image is None or model is None:
-        return None
-    
-    try:
-        model = model.to(self.device)
-        model.eval()
-        
-        original_size = content_image.size
-        
-        # This maintains the full image but may distort aspect ratio
-        transform = transforms.Compose([
-            transforms.Resize((256, 256)),  # Resize to exact dimensions
-            transforms.ToTensor(),
-            transforms.Normalize(mean=[0.485, 0.456, 0.406],
-                                 std=[0.229, 0.224, 0.225])
-        ])
-        
-        
-        content_tensor = transform(content_image).unsqueeze(0).to(self.device)
-        style_tensor = transform(style_image).unsqueeze(0).to(self.device)
+        """Apply AdaIN-based style transfer"""
+        if content_image is None or style_image is None or model is None:
+            return None
         
-        with torch.no_grad():
-            output = model(content_tensor, style_tensor, alpha=alpha)
+        try:
+            model = model.to(self.device)
+            model.eval()
             
-            # Denormalize
-            output = output.squeeze(0).cpu()
-            output = output * torch.tensor([0.229, 0.224, 0.225]).view(3, 1, 1)
-            output = output + torch.tensor([0.485, 0.456, 0.406]).view(3, 1, 1)
-            output = torch.clamp(output, 0, 1)
+            original_size = content_image.size
             
-            # Convert to PIL
-            output_img = transforms.ToPILImage()(output)
-            output_img = output_img.resize(original_size, Image.LANCZOS)
-        
-        return output_img
-    
-    except Exception as e:
-        print(f"Error applying AdaIN style: {e}")
-        traceback.print_exc()
-        return None
+            # This maintains the full image but may distort aspect ratio
+            transform = transforms.Compose([
+                transforms.Resize((256, 256)),  # Resize to exact dimensions
+                transforms.ToTensor(),
+                transforms.Normalize(mean=[0.485, 0.456, 0.406],
+                                     std=[0.229, 0.224, 0.225])
+            ])
+            
+            
+            content_tensor = transform(content_image).unsqueeze(0).to(self.device)
+            style_tensor = transform(style_image).unsqueeze(0).to(self.device)
+            
+            with torch.no_grad():
+                output = model(content_tensor, style_tensor, alpha=alpha)
+                
+                # Denormalize
+                output = output.squeeze(0).cpu()
+                output = output * torch.tensor([0.229, 0.224, 0.225]).view(3, 1, 1)
+                output = output + torch.tensor([0.485, 0.456, 0.406]).view(3, 1, 1)
+                output = torch.clamp(output, 0, 1)
+                
+                # Convert to PIL
+                output_img = transforms.ToPILImage()(output)
+                output_img = output_img.resize(original_size, Image.LANCZOS)
+            
+            return output_img
         
+        except Exception as e:
+            print(f"Error applying AdaIN style: {e}")
+            traceback.print_exc()
+            return None
+            
 # ===========================
 # HELPER FUNCTIONS
 # ===========================