Update app.py

app.py

@@ -7,10 +7,9 @@ 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 = """ """
 
@@ -29,8 +28,6 @@ MAX_SEED = np.iinfo(np.int32).max
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU, This may not work on CPU.</p>"
 
-MAX_SEED = np.iinfo(np.int32).max
-
 USE_TORCH_COMPILE = 0
 ENABLE_CPU_OFFLOAD = 0
 
@@ -40,25 +37,21 @@ style_list = [
         "prompt": "{prompt}",
         "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": "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": "Cinematic",
         "prompt": "cinematic still {prompt} . emotional, harmonious, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy",
@@ -84,15 +77,13 @@ style_list = [
         "prompt": "concept art {prompt} . digital artwork, illustrative, painterly, matte painting, highly detailed",
         "negative_prompt": "photo, photorealistic, realism, ugly",
     },
-
     {
         "name": "3D Model",
         "prompt": "professional 3d model {prompt} . octane render, highly detailed, volumetric, dramatic lighting",
         "negative_prompt": "ugly, deformed, noisy, low poly, blurry, painting",
     },
+]
 
-    
-]   
 styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "(No style)"
@@ -103,7 +94,6 @@ 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 = "",
@@ -118,11 +108,10 @@ def stab(
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ):
-
     seed = int(randomize_seed_fn(seed, randomize_seed))
 
     if not use_negative_prompt:
-        negative_prompt = ""  # type: ignore
+        negative_prompt = ""
     prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
 
     prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", variant="bf16", torch_dtype=torch.bfloat16)
@@ -169,6 +158,7 @@ footer {
     visibility: hidden
 }
 '''
+
 with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
     gr.Markdown(DESCRIPTION)
     gr.DuplicateButton(
@@ -179,7 +169,7 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
 
     with gr.Group():
         with gr.Row():
-            prompt = gr.Text(
+            prompt = gr.Textbox(
                 label="Prompt",
                 show_label=False,
                 max_lines=1,
@@ -190,14 +180,13 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
         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.Text(
-                label="Negative prompt",
-                max_lines=5,
-                line=4,
-                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,
-            )
+        negative_prompt = gr.Textbox(
+            label="Negative prompt",
+            max_lines=5,
+            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():
             num_inference_steps = gr.Slider(
                 label="Steps",
@@ -247,15 +236,14 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
                 value=6,
             )
     with gr.Row(visible=True):
-            style_selection = gr.Radio(
-                show_label=True,
-                container=True,
-                interactive=True,
-                choices=STYLE_NAMES,
-                value=DEFAULT_STYLE_NAME,
-                label="Image Style",
-            )
-        
+        style_selection = gr.Radio(
+            show_label=True,
+            container=True,
+            interactive=True,
+            choices=STYLE_NAMES,
+            value=DEFAULT_STYLE_NAME,
+            label="Image Style",
+        )
 
     gr.Examples(
         examples=examples,
@@ -271,7 +259,6 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
         outputs=negative_prompt,
         api_name=False,
     )
-    
 
     gr.on(
         triggers=[
@@ -296,6 +283,6 @@ with gr.Blocks(css=css, theme="xiaobaiyuan/theme_brief") as demo:
         outputs=[result, seed],
         api_name="run",
     )
-    
+
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(show_api=False, debug=False, share=True)