Update boomerang.py

boomerang.py

@@ -1,6 +1,6 @@
 import inspect
+import os
 from PIL import Image
-import streamlit as st
 import torch
 from torch import autocast
 from torchvision import transforms as T
@@ -11,7 +11,7 @@ from diffusers import StableDiffusionPipeline
 from diffusers.models.unet_2d_condition import UNet2DConditionOutput
 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
 
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=st.secrets['HF_TOKEN_SD'])  
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=os.environ.get('HF_TOKEN_SD'))  
 #pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True)  
 
 pipe = pipe.to('cuda')