Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import torch
 from PIL import Image
 import utils
+import streamlit as st
 
 is_colab = utils.is_google_colab()
 
@@ -10,7 +11,9 @@ scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="sca
                           num_train_timesteps=1000, clip_sample=False, set_alpha_to_one=False)
 
 model_id_or_path = "CompVis/stable-diffusion-v1-4"
-pipe = CycleDiffusionPipeline.from_pretrained(model_id_or_path, scheduler=scheduler).to("cuda")
+pipe = CycleDiffusionPipeline.from_pretrained(model_id_or_path,
+                                              use_auth_token=st.secrets["USER_TOKEN"],
+                                              scheduler=scheduler)
 
 if torch.cuda.is_available():
     pipe = pipe.to("cuda")