调整项目结构

app.py

@@ -6,10 +6,17 @@ import random
 import logging
 import utils
 from diffusers.models import AutoencoderKL
+from config import (
+    MODEL,
+    MIN_IMAGE_SIZE,
+    MAX_IMAGE_SIZE,
+    DEFAULT_NEGATIVE_PROMPT,
+    scheduler_list,
+)
+
 
 MAX_SEED = np.iinfo(np.int32).max
-MIN_IMAGE_SIZE = 512
-MAX_IMAGE_SIZE = 2048
+
 
 # Enhanced logging configuration
 logging.basicConfig(
@@ -35,11 +42,11 @@ if torch.cuda.is_available():
             "madebyollin/sdxl-vae-fp16-fix",
             torch_dtype=torch.float16,
         )
-        pipe = utils.load_pipeline("cagliostrolab/animagine-xl-4.0", device, vae=vae)
+        pipe = utils.load_pipeline(MODEL, device, vae=vae)
         logger.info("Pipeline loaded successfully on GPU!")
     except Exception as e:
         logger.error(f"Error loading VAE, falling back to default: {e}")
-        pipe = utils.load_pipeline("cagliostrolab/animagine-xl-4.0", device)
+        pipe = utils.load_pipeline(MODEL, device)
 else:
     logger.warning("CUDA not available, running on CPU")
     pipe = None
@@ -82,15 +89,6 @@ def generate(
 
 
 
-scheduler_list = [
-    "DPM++ 2M Karras",
-    "DPM++ SDE Karras",
-    "DPM++ 2M SDE Karras",
-    "Euler",
-    "Euler a",
-    "DDIM"
-]
-
 
 title = "# Animagine XL 4.0 Demo"
 
@@ -120,13 +118,14 @@ with gr.Blocks(css=custom_css).queue() as demo:
             with gr.Column():
                 prompt = gr.Text(
                     label="Prompt",
-                    max_lines=1,
+                    max_lines=5,
                     placeholder="Enter your prompt",
                 )
                 negative_prompt = gr.Text(
                     label="Negative prompt",
-                    max_lines=1,
+                    max_lines=5,
                     placeholder="Enter a negative prompt",
+                    value=DEFAULT_NEGATIVE_PROMPT,
                 )
             with gr.Row():
                 width = gr.Slider(

config.py

@@ -0,0 +1,13 @@
+MODEL = "cagliostrolab/animagine-xl-4.0"
+DEFAULT_NEGATIVE_PROMPT = "lowres, bad anatomy, bad hands, text, error, missing finger, extra digits, fewer digits, cropped, worst quality, low quality, low score, bad score, average score, signature, watermark, username, blurry"
+MIN_IMAGE_SIZE = 512
+MAX_IMAGE_SIZE = 2048
+
+scheduler_list = [
+    "DPM++ 2M Karras",
+    "DPM++ SDE Karras",
+    "DPM++ 2M SDE Karras",
+    "Euler",
+    "Euler a",
+    "DDIM"
+]