Update app.py

app.py

@@ -7,9 +7,7 @@ Original file is located at
     https://colab.research.google.com/drive/19xx6Nu4FeiGj-TzTUFxBf-15IkeuFx_F
 """
 
-import os
-USER_TOKEN = os.environ.getattribute("USER_TOKEN")
-
+import streamlit as st
 import gradio as gr
 import torch as th
 
@@ -31,7 +29,7 @@ print(device)
 # init stable diffusion model
 pipe = StableDiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
-    use_auth_token=USER_TOKEN
+    use_auth_token=st.secrets["USER_TOKEN"]
 ).to(device)