Spaces:
Runtime error
Runtime error
DiningSystem
commited on
Commit
•
eeb353a
1
Parent(s):
5153ff7
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ my_file.close()
|
|
17 |
|
18 |
hf.login(token=os.environ['model_token'])
|
19 |
#remember to login with token before loading model
|
20 |
-
def text_to_hair(prompt, guidance_scale=8, num_inference_steps=30, styles=data_into_list, model_path ="
|
21 |
|
22 |
pipe = StableDiffusionPipeline.from_pretrained(os.environ['bmd'], torch_dtype=torch.float16, use_auth_token=True)
|
23 |
pipe.unet.load_attn_procs(model_path)
|
|
|
17 |
|
18 |
hf.login(token=os.environ['model_token'])
|
19 |
#remember to login with token before loading model
|
20 |
+
def text_to_hair(prompt, guidance_scale=8, num_inference_steps=30, styles=data_into_list, model_path ="DiningSystem/hair-model2"):
|
21 |
|
22 |
pipe = StableDiffusionPipeline.from_pretrained(os.environ['bmd'], torch_dtype=torch.float16, use_auth_token=True)
|
23 |
pipe.unet.load_attn_procs(model_path)
|