Update app.py

app.py

@@ -15,9 +15,12 @@ import torch
 # API Key for weather
 API_KEY = os.getenv("Weather_Token")
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+hf_token = os.getenv("HF_TOKEN")
+login(token=hf_token)
+
 sd_pipe = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
+    use_auth_token=hf_token,
     torch_dtype=torch.float16 if device == "cuda" else torch.float32
 ).to(device)