Update app.py

app.py

@@ -59,8 +59,8 @@ def generate(
     if torch.cuda.is_available():
 
         if not use_vae:  
-        pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
-        pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
+            pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)