Manjushri commited on
Commit
995afa9
·
1 Parent(s): 58c32b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -5,8 +5,10 @@ import modin.pandas as pd
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
- login(token='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')
9
 
 
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-0.9")
12
  pipe = pipe.to(device)
 
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
+ import os
9
 
10
+ os.environ.get("HF_KEY")
11
+ login(token='HF_KEY')
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-0.9")
14
  pipe = pipe.to(device)