Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
|
3 |
import torch
|
4 |
from PIL import Image
|
5 |
|
6 |
-
model_id = "alibaba-pai/pai-diffusion-
|
7 |
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
8 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
9 |
pipe = pipe.to("cuda")
|
@@ -14,9 +14,9 @@ def infer_text2img(prompt, guide, steps):
|
|
14 |
|
15 |
with gr.Blocks() as demo:
|
16 |
examples = [
|
17 |
-
["
|
18 |
-
["
|
19 |
-
["
|
20 |
]
|
21 |
with gr.Row():
|
22 |
with gr.Column(scale=1, ):
|
|
|
3 |
import torch
|
4 |
from PIL import Image
|
5 |
|
6 |
+
model_id = "alibaba-pai/pai-diffusion-artist-xlarge-zh"
|
7 |
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
8 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
9 |
pipe = pipe.to("cuda")
|
|
|
14 |
|
15 |
with gr.Blocks() as demo:
|
16 |
examples = [
|
17 |
+
["草地上的帐篷,背景是山脉"],
|
18 |
+
["卧室里有一张床和一张桌子"],
|
19 |
+
["雾蒙蒙的日出在湖面上"],
|
20 |
]
|
21 |
with gr.Row():
|
22 |
with gr.Column(scale=1, ):
|