Liyulingyue commited on
Commit
2f81857
·
1 Parent(s): a5bdce9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -10,11 +10,12 @@ from ppdiffusers import StableDiffusionPipeline
10
  import git
11
 
12
  # 获取模型参数
13
- repo = git.Repo.clone_from(url='https://huggingface.co/Liyulingyue/Neolle_Face_Generator', to_path="./dream_outputs")
14
-
15
  # 加载模型
16
- model_path = "dream_outputs"
17
- pipe = StableDiffusionPipeline.from_pretrained(model_path)
 
 
18
 
19
  def generate_images(prompt, num_inference_steps, guidance_scale):
20
  # num_inference_steps to number
 
10
  import git
11
 
12
  # 获取模型参数
13
+ # repo = git.Repo.clone_from(url='https://huggingface.co/Liyulingyue/Neolle_Face_Generator', to_path="./dream_outputs")
 
14
  # 加载模型
15
+ # model_path = "dream_outputs"
16
+ # pipe = StableDiffusionPipeline.from_pretrained(model_path)
17
+
18
+ pipe = StableDiffusionPipeline.from_pretrained('Liyulingyue/Neolle_Face_Generator', from_hf_hub=True)
19
 
20
  def generate_images(prompt, num_inference_steps, guidance_scale):
21
  # num_inference_steps to number