Update app.py

app.py

@@ -5,8 +5,8 @@ import gradio as gr
 import torch
 
 SEED = 42
-AUTH_TOKEN = os.getenv("auth_token")
-HF_TOKEN = os.getenv('HF_TOKEN')
+AUTH_TOKEN = os.environ.get("auth_token")
+HF_TOKEN = os.environ.get("HF_TOKEN")
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=AUTH_TOKEN)