Update app.py

app.py

@@ -7,9 +7,10 @@ import uuid
 import gradio as gr
 import numpy as np
 from PIL import Image
+import spaces
+from typing import Tuple
 import torch
 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
-from typing import Tuple
 
 DESCRIPTION = """ """
 
@@ -38,19 +39,19 @@ style_list = [
         "negative_prompt": "",
     },
     {
-        "name": "3840 x 2160",
-        "prompt": "hyper-realistic 8K image of {prompt} . ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
+    "name": "3840 x 2160",
+    "prompt": "hyper-realistic 8K image of {prompt} . ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
+    "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
     },
     {
-        "name": "2560 × 1440",
-        "prompt": "hyper-realistic 4K image of {prompt} . ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
+    "name": "2560 × 1440",
+    "prompt": "hyper-realistic 4K image of {prompt} . ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
+    "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
     },
     {
-        "name": "HDR",
-        "prompt": "HDR photo of {prompt} . high dynamic range, vivid colors, sharp contrast, realistic, detailed, high resolution, professional",
-        "negative_prompt": "dull, low contrast, blurry, unrealistic, cartoonish, ugly, deformed",
+    "name": "HDR",
+    "prompt": "HDR photo of {prompt} . high dynamic range, vivid colors, sharp contrast, realistic, detailed, high resolution, professional",
+    "negative_prompt": "dull, low contrast, blurry, unrealistic, cartoonish, ugly, deformed",
     },
     {
         "name": "Cinematic",
@@ -94,6 +95,7 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
         negative = ""
     return p.replace("{prompt}", positive), n + negative
 
+@spaces.GPU(enable_queue=True)
 def stab(
     prompt: str,
     negative_prompt: str = "",
@@ -169,7 +171,7 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
 
     with gr.Group():
         with gr.Row():
-            prompt = gr.Textbox(
+            prompt = gr.Text(
                 label="Prompt",
                 show_label=False,
                 max_lines=1,
@@ -179,12 +181,11 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
             run_button = gr.Button("Run")
         result = gr.Gallery(label="Result", columns=1, preview=True)
     with gr.Accordion("Advanced options", open=False):
-        use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
-        negative_prompt = gr.Textbox(
+        use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=False, visible=True)
+        negative_prompt = gr.Text(
             label="Negative prompt",
-            max_lines=5,
+            max_lines=1,
             placeholder="Enter a negative prompt",
-            value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
             visible=True,
         )
         with gr.Row():