Update app.py

app.py

@@ -8,9 +8,7 @@ import spaces
 import torch
 from huggingface_hub import login, whoami
 
-# ─────────────────────────────────────────────
-# 全局配置
-# ─────────────────────────────────────────────
+
 HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
 DEFAULT_PIPELINE_PATH = "black-forest-labs/FLUX.1-dev"
@@ -22,7 +20,6 @@ MAX_IMAGE_SIZE = 2048
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
-# 认证状态
 auth_status = "🔴 Not Authenticated"
 if HF_TOKEN:
     try:
@@ -34,14 +31,9 @@ if HF_TOKEN:
         logging.error(f"HF authentication failed: {e}")
         auth_status = f"🔴 Authentication Error: {str(e)}"
 
-# ─────────────────────────────────────────────
-# 在 GPU 子进程 import 阶段就把大模型读进显存
-# 只在 GPU 进程执行，CPU 主进程跳过
-# ─────────────────────────────────────────────
 
-# 检测是否在 GPU 环境中运行
+
 def is_gpu_available():
-    """检测当前环境是否支持 GPU"""
     try:
         import torch
         return torch.cuda.is_available()
@@ -84,11 +76,7 @@ if is_gpu_available():
     
     print("✅ [GPU init] All models loaded successfully!")
 
-# ─────────────────────────────────────────────
-# Qwen 提示词增强函数
-# ─────────────────────────────────────────────
 def enhance_prompt_with_qwen(original_prompt):
-    """使用预加载的 Qwen 模型增强提示词"""
     if not is_gpu_available():
         return original_prompt
         
@@ -141,7 +129,7 @@ Elaborate on each core requirement to create a rich description.
         with torch.no_grad():
             generated_ids = QWEN_MODEL.generate(
                 **model_inputs,
-                max_new_tokens=512,          # ← 从 4096 改成 512 够用
+                max_new_tokens=512,         
                 temperature=0.6,
                 top_p=0.9,
                 do_sample=True,
@@ -165,9 +153,7 @@ Elaborate on each core requirement to create a rich description.
         logging.error(f"Qwen enhancement failed: {e}")
         return original_prompt
 
-# ─────────────────────────────────────────────
-# 主要生成函数（使用预加载的模型）
-# ─────────────────────────────────────────────
+
 @spaces.GPU(duration=300)
 def generate_poster(
     original_prompt, 
@@ -221,9 +207,7 @@ def generate_poster(
         logging.error(f"Generation failed: {e}")
         return None, f"❌ Generation failed: {str(e)}", ""
 
-# ─────────────────────────────────────────────
-# Gradio Interface
-# ─────────────────────────────────────────────
+
 def create_interface():
     """Create Gradio interface"""
     
@@ -257,7 +241,7 @@ def create_interface():
 
             with gr.Column(scale=1):
                 gr.Markdown("### 2. Results")
-                image_output = gr.Image(label="Generated Image", type="pil", show_download_button=True, height=340, container=False, elem_classes=["preserve-aspect-ratio"])
+                image_output = gr.Image(label="Generated Image", type="pil", show_download_button=True, height=350, container=False, elem_classes=["preserve-aspect-ratio"])
                 recapped_prompt_output = gr.Textbox(label="Final Prompt Used", lines=5, interactive=False)
                 status_output = gr.Textbox(label="Status Log", lines=4, interactive=False)
 
@@ -271,9 +255,6 @@ def create_interface():
     
     return demo
 
-# ─────────────────────────────────────────────
-# 启动应用
-# ─────────────────────────────────────────────
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch(