Update app.py

app.py

@@ -1,35 +1,31 @@
-import torch
-from diffusers import StableDiffusionXLPipeline
+import os
 from huggingface_hub import HfApi, HfFolder, upload_folder
+from diffusers import StableDiffusionXLPipeline
+import torch
 
-# 🔹 Base model your LoRA was trained on
 base_model = "stabilityai/stable-diffusion-xl-base-1.0"
-
-# 🔹 Your LoRA model on Hugging Face
 lora_model_id = "CoolKrishh/Comic-SDXL-LoRA"
-
-# 🔹 Output directory for the merged pipeline
 output_dir = "comic-sdxl-full"
 
 print("⬇️ Loading base model...")
 pipe = StableDiffusionXLPipeline.from_pretrained(base_model, torch_dtype=torch.float16)
-
-print("🔗 Attaching LoRA weights...")
+print("🔗 Loading LoRA weights...")
 pipe.load_lora_weights(lora_model_id)
 
 print("💾 Saving merged pipeline...")
 pipe.save_pretrained(output_dir)
 
-print("✅ Saved full pipeline locally!")
+# ✅ Use token from environment
+token = os.getenv("HF_TOKEN")
 
-# 🔹 Upload merged model to your HF account
-api = HfApi()
-token = HfFolder.get_token()
+if not token:
+    raise ValueError("❌ No HF_TOKEN found. Add it in your Space secrets.")
 
-repo_id = "CoolKrishh/Comic-SDXL-Full"  # change name if needed
+api = HfApi()
+repo_id = "CoolKrishh/Comic-SDXL-Full"
 
 print("☁️ Uploading to Hugging Face Hub...")
-api.create_repo(repo_id=repo_id, repo_type="model", private=False, exist_ok=True)
+api.create_repo(repo_id=repo_id, repo_type="model", private=False, exist_ok=True, token=token)
 upload_folder(folder_path=output_dir, repo_id=repo_id, repo_type="model", token=token)
 
-print(f"🚀 Uploaded full pipeline to https://huggingface.co/{repo_id}")
+print(f"🚀 Uploaded successfully to https://huggingface.co/{repo_id}")