Manjushri commited on
Commit
cec24a1
·
1 Parent(s): 6bce7cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -6,8 +6,11 @@ from PIL import Image
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
  import streamlit as st
9
- from streamlit import secrets
10
- token=st.secrets(['HF_KEY'])
 
 
 
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-0.9")
13
  pipe = pipe.to(device)
 
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
  import streamlit as st
9
+ import os
10
+
11
+ #os.environ['HF_KEY']
12
+
13
+ token=st.secrets['HF_KEY']
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-0.9")
16
  pipe = pipe.to(device)