Update app.py

app.py

@@ -1,12 +1,15 @@
+   import os
    import gradio as gr
    from huggingface_hub import snapshot_download
-   import os
    import torch
    from diffusers import FluxPipeline
 
-   # 下载模型 (Space 启动时自动跑)
+   # 安装依赖 (Space 启动时自动跑)
+   os.system("pip install huggingface-hub transformers diffusers accelerate safetensors torch gradio")
+
+   # 下载模型
    model_dir = "models/postercraft/postercraft_rl"
-   snapshot_download(repo_id="PosterCraft/PosterCraft-v1_RL", local_dir=model_dir, resume_download=True)
+   snapshot_download(repo_id="PosterCraft/PosterCraft-v1_RL", local_dir=model_dir, resume_download=True, token=os.environ.get("HF_TOKEN"))  # 加 Token 支持
 
    # 加载模型
    pipe = FluxPipeline.from_pretrained(model_dir)
@@ -14,8 +17,11 @@
 
    # 生成函数
    def generate_poster(prompt):
-       image = pipe(prompt).images[0]
-       return image
+       try:
+           image = pipe(prompt).images[0]
+           return image
+       except Exception as e:
+           return f"Error: {str(e)}"  # 错误处理
 
    # Gradio UI
    iface = gr.Interface(