Update app.py

app.py

@@ -55,18 +55,19 @@ def generate(
     model = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
     vaecall = 'stabilityai/sd-vae-ft-mse',
     lora = 'amazonaws-la/juliette',
+    sampler = 'EulerAncestralDiscreteScheduler',
     lora_scale: float = 0.7,
 ) -> PIL.Image.Image:
     if torch.cuda.is_available():
 
         if not use_vae:  
             pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
-            pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+            pipe.scheduler = sampler.from_config(pipe.scheduler.config)
             
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
             pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
-            pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+            pipe.scheduler = sampler.from_config(pipe.scheduler.config)
             
         if use_lora:
             pipe.load_lora_weights(lora)
@@ -142,6 +143,7 @@ with gr.Blocks(css="style.css") as demo:
         visible=os.getenv("SHOW_DUPLICATE_BUTTON") == "1",
     )
     with gr.Group():
+        sampler = gr.Text(label='Sampler', value='EulerAncestralDiscreteScheduler')
         model = gr.Text(label='Modelo')
         vaecall = gr.Text(label='VAE')
         lora = gr.Text(label='LoRA')
@@ -330,6 +332,7 @@ with gr.Blocks(css="style.css") as demo:
             model,
             vaecall,
             lora,
+            sampler,
             lora_scale,
         ],
         outputs=result,