test model stabilityai/stable-diffusion-2

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
 
-# pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney")
-pipe = DiffusionPipeline.from_pretrained("admruul/anything-v3.0")
+# pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney") # about 750 s
+# pipe = DiffusionPipeline.from_pretrained("admruul/anything-v3.0") # about 2500 s
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
 
 # prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
 # image = pipe(prompt).images[0]