chywang commited on
Commit
3b9fac0
1 Parent(s): 6cfeb73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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-food-large-zh"
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, ):