Update app.py

app.py

@@ -157,7 +157,7 @@ def infer(
     print(enhanced_prompt)
     if latent_file:  # Check if a latent file is provided
         sd_image_a = torch.load(latent_file.name).to('cuda')  # Load the latent
-        sd_image_b = vae.encode(sd_image_a.to(torch.bfloat16)).latent_dist.sample().mul_(0.18215)
+        sd_image_b = pipe.vae.encode(sd_image_a.to(torch.bfloat16)).latent_dist.sample().mul_(0.18215)
         print("-- using latent file --")
         print('-- generating image --')
         with torch.no_grad():