Liyulingyue commited on
Commit
21db931
1 Parent(s): 752b1fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -6,11 +6,8 @@ import git
6
  # 获取模型参数
7
  repo = git.Repo.clone_from(url='https://huggingface.co/Liyulingyue/Neolle_Face_Generator', to_path="./dream_outputs")
8
 
9
- print(os.getcwd())
10
- print(os.listdir('.'))
11
-
12
  # 加载模型
13
- model_path = "/home/user/app/dream_outputs"
14
  pipe = StableDiffusionPipeline.from_pretrained(model_path)
15
 
16
  def generate_images(prompt="Neolle", num_inference_steps=50, guidance_scale=7.5):
 
6
  # 获取模型参数
7
  repo = git.Repo.clone_from(url='https://huggingface.co/Liyulingyue/Neolle_Face_Generator', to_path="./dream_outputs")
8
 
 
 
 
9
  # 加载模型
10
+ model_path = "dream_outputs"
11
  pipe = StableDiffusionPipeline.from_pretrained(model_path)
12
 
13
  def generate_images(prompt="Neolle", num_inference_steps=50, guidance_scale=7.5):