Spaces:
Runtime error
Runtime error
Commit
·
c72bdea
1
Parent(s):
5cd9952
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import huggingface_hub as hf
|
|
5 |
import os
|
6 |
hf.login(token=os.environ['model_token'])
|
7 |
#remember to login with token before loading model
|
8 |
-
def text_to_hair(prompt, num_inference_steps=
|
9 |
|
10 |
pipe = StableDiffusionPipeline.from_pretrained(os.environ['bmd'], torch_dtype=torch.float16)
|
11 |
pipe.unet.load_attn_procs(model_path)
|
|
|
5 |
import os
|
6 |
hf.login(token=os.environ['model_token'])
|
7 |
#remember to login with token before loading model
|
8 |
+
def text_to_hair(prompt, num_inference_steps=50, guidance_scale=6, model_path ="CVH-vn1210/hair-model"):
|
9 |
|
10 |
pipe = StableDiffusionPipeline.from_pretrained(os.environ['bmd'], torch_dtype=torch.float16)
|
11 |
pipe.unet.load_attn_procs(model_path)
|