update app

app.py

@@ -8,11 +8,14 @@ import subprocess
 from diffusers import StableDiffusionPipeline
 import time
 
+repo_id = "runwayml/stable-diffusion-v1-5"
+dpm = DPMSolverMultistepScheduler.from_pretrained(repo_id, subfolder="scheduler")
+
 model_id = "dicoo_model"
 
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=dpm, torch_dtype=torch.float)
 
-def predict(prompt, steps=30, seed=42, guidance_scale=7.5):
+def predict(prompt, steps=25, seed=42, guidance_scale=7.5):
     # cpu info
     # print(subprocess.check_output(["cat /proc/cpuinfo | grep 'model name' |uniq"], stderr=subprocess.STDOUT).decode("utf8"))
     print("prompt: ", prompt)
@@ -36,7 +39,7 @@ gr.Interface(
     predict,
     inputs=[
         gr.inputs.Textbox(label='Prompt', default='a lovely <dicoo> in red dress and hat, in the snowy and brightly night, with many brightly buildings'),
-        gr.inputs.Slider(1, 100, label='Inference Steps', default=30, step=1),
+        gr.inputs.Slider(1, 100, label='Inference Steps', default=25, step=1),
         gr.inputs.Slider(0, 2147483647, label='Seed', default=random_seed, step=1),
         gr.inputs.Slider(1.0, 20.0, label='Guidance Scale - how much the prompt will influence the results', default=6.0, step=0.1),
     ],