Update app.py

app.py

@@ -226,8 +226,14 @@ MAX_IMAGE_SIZE = 4096
 
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+    token = os.environ.get("HF_TOKEN")  # 从环境变量读取令牌
+    model_path = hf_hub_download(
+        repo_id="Menyu/Pixel",  # 模型仓库名称（非完整URL）
+        filename="MiaoMiaoPixel_V1.0.safetensors",
+        use_auth_token=token
+    )
     pipe = StableDiffusionXLPipeline.from_single_file(
-        "https://huggingface.co/bluepen5805/illustrious_pencil-XL/illustrious_pencil-XL-v2.0.0.safetensors",
+        model_path,
         vae=vae,
         use_safetensors=True,
         torch_dtype=torch.float16,
@@ -297,7 +303,7 @@ h1{text-align:center}
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("""# 梦羽的模型生成器
-        ### 快速生成illustrious_pencil-XL模型的图片""")
+        ### 快速生成MiaoMiaoPixel 像素模型的图片""")
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(