k36 commited on
Commit
6e2b223
1 Parent(s): cea75e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,11 +15,11 @@ def select_model(choice):#用于选择模型
15
  pipe = None
16
  paddle.device.cuda.empty_cache() #释放GPU显存
17
  if choice == '文本引导图片风格变化':
18
- pipe1 = StableDiffusionImg2ImgPipeline.from_pretrained("/home/aistudio/model1")
19
  elif choice == '文本引导图像区域生成':
20
- pipe1 = StableDiffusionInpaintPipeline.from_pretrained("/home/aistudio/model2")
21
  elif choice == '文图引导图像区域生成':
22
- pipe1 = PaintByExamplePipeline.from_pretrained("/home/aistudio/model3")
23
  pipe = pipe1
24
  return pipe1 #用于输出导入的模型相关信息
25
 
 
15
  pipe = None
16
  paddle.device.cuda.empty_cache() #释放GPU显存
17
  if choice == '文本引导图片风格变化':
18
+ pipe1 = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
19
  elif choice == '文本引导图像区域生成':
20
+ pipe1 = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting")
21
  elif choice == '文图引导图像区域生成':
22
+ pipe1 = PaintByExamplePipeline.from_pretrained("Fantasy-Studio/Paint-by-Example")
23
  pipe = pipe1
24
  return pipe1 #用于输出导入的模型相关信息
25