Update app.py

app.py

@@ -1,13 +1,14 @@
    import gradio as gr
    from huggingface_hub import snapshot_download
    import os
+   import torch
    from diffusers import FluxPipeline
 
    # 下载模型 (Space 启动时自动跑)
    model_dir = "models/postercraft/postercraft_rl"
    snapshot_download(repo_id="PosterCraft/PosterCraft-v1_RL", local_dir=model_dir, resume_download=True)
 
-   # 加载模型 (替换为您的加载逻辑)
+   # 加载模型
    pipe = FluxPipeline.from_pretrained(model_dir)
    pipe.to("cuda" if torch.cuda.is_available() else "cpu")