Update app.py

app.py

@@ -93,7 +93,7 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
     
     generator = torch.Generator(device=device.value).manual_seed(seed)
     latents = torch.randn(
-        (1, unet.in_channels, 512 // 8, 512 // 8),
+        (1, unet.value.in_channels, 512 // 8, 512 // 8),
         generator = generator,
         device = device.value
     ).bfloat16()
@@ -152,7 +152,7 @@ def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, st
    
     generator = torch.Generator(device=device.value).manual_seed(seed)
     latents = torch.randn(
-        (1, unet.in_channels, 512 // 8, 512 // 8),
+        (1, unet.value.in_channels, 512 // 8, 512 // 8),
         generator = generator,
         device = device.value
     ).bfloat16()