Manjushri commited on
Commit
1be771e
·
1 Parent(s): 9b18847

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,9 +5,9 @@ import modin.pandas as pd
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
- import os
9
 
10
- token='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)
 
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
+ import streamlit as st
9
 
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)