DiningSystem commited on
Commit
9c64075
1 Parent(s): 6ec1d60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,8 +1,7 @@
1
  import gradio as gr
2
  from diffusers import StableDiffusionPipeline
3
  import torch
4
- import huggingface_hub as hf
5
- hf.login(token="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")
6
  #remember to login with token before loading model
7
  def text_to_hair(prompt, num_inference_steps=200, guidance_scale=9, model_path ="./"):
8
  pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", torch_dtype=torch.float16, use_auth_token=True)
 
1
  import gradio as gr
2
  from diffusers import StableDiffusionPipeline
3
  import torch
4
+ #import huggingface_hub as hf
 
5
  #remember to login with token before loading model
6
  def text_to_hair(prompt, num_inference_steps=200, guidance_scale=9, model_path ="./"):
7
  pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", torch_dtype=torch.float16, use_auth_token=True)