Update app.py

app.py

@@ -24,7 +24,6 @@ CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1")
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "6000"))
 USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
-PORT = int(os.getenv("DEMO_PORT", "15432"))
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -88,6 +87,12 @@ STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "(No style)"
 NUM_IMAGES_PER_PROMPT = 1
 
+def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
+    p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
+    if not negative:
+        negative = ""
+    return p.replace("{prompt}", positive), n + negative
+
 if torch.cuda.is_available():
 
     pipe = PixArtAlphaPipeline.from_pretrained(
@@ -141,7 +146,7 @@ def generate(
     generator = torch.Generator().manual_seed(seed)
 
 if not use_negative_prompt:
-        negative_prompt = None  # type: ignore
+        negative_prompt = None
     prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
 
     images = pipe(