Update app.py

app.py

@@ -1,11 +1,12 @@
 from diffusers import StableDiffusionPipeline
-from peft import PeftModel  # PEFT 支持
+from huggingface_hub import hf_hub_download
 import torch
 import gradio as gr
 
-# 模型和 LoRA 權重的 URL
+# 模型和 LoRA 存儲庫信息
 base_model = "stabilityai/stable-diffusion-xl-base-1.0"  # 基礎模型
-lora_weights_url = "https://huggingface.co/hyder133/chiikawa_stype/resolve/main/tkw1.safetensors"
+repo_id = "hyder133/chiikawa_stype"                     # Hugging Face 存儲庫 ID
+lora_filename = "tkw1.safetensors"                      # LoRA 權重文件名
 
 # 嘗試加載到 GPU，如果不可用則回退到 CPU
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -15,8 +16,9 @@ dtype = torch.float16 if device == "cuda" else torch.float32
 pipe = StableDiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
 pipe = pipe.to(device)
 
-# 加載 LoRA 權重
-pipe.load_lora_weights(lora_weights_url)
+# 下載並加載 LoRA 權重
+lora_file_path = hf_hub_download(repo_id=repo_id, filename=lora_filename)
+pipe.load_lora_weights(lora_file_path)
 
 # 測試生成函數
 def generate_image(prompt, width, height, steps, guidance_scale):