Update app.py

app.py

@@ -20,7 +20,7 @@ import numpy as np
 import pandas as pd
 import plotly.graph_objects as go
 import torch
-from diffusers import DPMSolverMultistepScheduler, StableDiffusionPipeline
+from diffusers import StableDiffusionPipeline, UniPCMultistepScheduler
 from sklearn.decomposition import PCA
 
 # On ZeroGPU Spaces the `spaces` package provides a GPU decorator.
@@ -52,7 +52,7 @@ if not USE_SAFETY_CHECKER:
     _pipe_kwargs.update(safety_checker=None, requires_safety_checker=False)
 
 pipe = StableDiffusionPipeline.from_pretrained(MODEL_ID, **_pipe_kwargs)
-pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 pipe = pipe.to(DEVICE)
 pipe.set_progress_bar_config(disable=True)
 
@@ -393,7 +393,7 @@ with gr.Blocks(title="Text Embedding Explorer for Stable Diffusion 1.5") as demo
 
     with gr.Accordion("Generation settings (shared by all tabs)", open=False):
         with gr.Row():
-            steps = gr.Slider(10, 40, value=20, step=1, label="Steps")
+            steps = gr.Slider(8, 48, value=16, step=1, label="Steps")
             cfg = gr.Slider(1, 15, value=7.5, step=0.5, label="Guidance scale")
             size = gr.Dropdown([384, 448, 512], value=512, label="Image size (px)")
         with gr.Row():