Update app.py

app.py

@@ -1,20 +1,26 @@
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
-from huggingface_hub import hf_hub_download
+import os
+from huggingface_hub import login
+
+# Login using the token
+login(token=os.environ.get("HF_TOKEN"))
 
 # Initialize the pipeline with CPU
 model_id = "runwayml/stable-diffusion-v1-5"
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,
     torch_dtype=torch.float32,
+    use_auth_token=os.environ.get("HF_TOKEN")
 )
 
 # Load your LoRA weights
 lora_path = "naonauno/40k-half-sd15"
 pipe.load_lora_weights(
     lora_path,
-    weight_name="40kHalf.safetensors"
+    weight_name="40kHalf.safetensors",
+    use_auth_token=os.environ.get("HF_TOKEN")
 )
 
 # Set the cross attention processor scale