Update app.py

app.py

@@ -232,7 +232,7 @@ def generate_with_prompt_style_guidance(prompt, style, seed=42):
 
     # Prep latents
     latents = torch.randn(
-    (batch_size, unet.in_channels, height // 8, width // 8),
+    (batch_size, unet.config.in_channels, height // 8, width // 8),
     generator=generator,
     )
     latents = latents.to(torch_device)