Update

README.md

@@ -4,7 +4,7 @@ emoji: 🚀
 colorFrom: blue
 colorTo: purple
 sdk: gradio
-sdk_version: 3.42.0
+sdk_version: 3.43.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -21,6 +21,11 @@ if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
 style_list = [
+    {
+        "name": "(No style)",
+        "prompt": "{prompt}",
+        "negative_prompt": "",
+    },
     {
         "name": "Cinematic",
         "prompt": "cinematic still {prompt} . emotional, harmonious, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy",
@@ -56,16 +61,25 @@ style_list = [
         "prompt": "ethereal fantasy concept art of  {prompt} . magnificent, celestial, ethereal, painterly, epic, majestic, magical, fantasy art, cover art, dreamy",
         "negative_prompt": "photographic, realistic, realism, 35mm film, dslr, cropped, frame, text, deformed, glitch, noise, noisy, off-center, deformed, cross-eyed, closed eyes, bad anatomy, ugly, disfigured, sloppy, duplicate, mutated, black and white",
     },
+    {
+        "name": "Neonpunk",
+        "prompt": "neonpunk style {prompt} . cyberpunk, vaporwave, neon, vibes, vibrant, stunningly beautiful, crisp, detailed, sleek, ultramodern, magenta highlights, dark purple shadows, high contrast, cinematic, ultra detailed, intricate, professional",
+        "negative_prompt": "painting, drawing, illustration, glitch, deformed, mutated, cross-eyed, ugly, disfigured",
+    },
+    {
+        "name": "Manga",
+        "prompt": "manga style {prompt} . vibrant, high-energy, detailed, iconic, Japanese comic style",
+        "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, Western comic style",
+    },
 ]
 
 styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
-default_style_name = "Photographic"
-default_style = styles[default_style_name]
-style_names = list(styles.keys())
+STYLE_NAMES = list(styles.keys())
+DEFAULT_STYLE_NAME = "(No style)"
 
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str, str]:
-    p, n = styles.get(style_name, default_style)
+    p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     return p.replace("{prompt}", positive), n + negative
 
 
@@ -101,12 +115,13 @@ def run(
     image: PIL.Image.Image,
     prompt: str,
     negative_prompt: str,
-    style_name: str = default_style_name,
+    style_name: str = DEFAULT_STYLE_NAME,
     num_steps: int = 25,
     guidance_scale: float = 5,
     adapter_conditioning_scale: float = 0.8,
-    cond_tau: float = 0.8,
+    adapter_conditioning_factor: float = 0.8,
     seed: int = 0,
+    progress=gr.Progress(track_tqdm=True),
 ) -> PIL.Image.Image:
     image = image.convert("RGB")
     image = TF.to_tensor(image) > 0.5
@@ -123,7 +138,7 @@ def run(
         generator=generator,
         guidance_scale=guidance_scale,
         adapter_conditioning_scale=adapter_conditioning_scale,
-        cond_tau=cond_tau,
+        adapter_conditioning_factor=adapter_conditioning_factor,
     ).images[0]
     return out
 
@@ -150,10 +165,13 @@ with gr.Blocks(css="style.css") as demo:
                     height=600,
                 )
                 prompt = gr.Textbox(label="Prompt")
+                style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
                 run_button = gr.Button("Run")
             with gr.Accordion("Advanced options", open=False):
-                style = gr.Dropdown(choices=style_names, value=default_style_name, label="Style")
-                negative_prompt = gr.Textbox(label="Negative prompt")
+                negative_prompt = gr.Textbox(
+                    label="Negative prompt",
+                    value=" extra digit, fewer digits, cropped, worst quality, low quality, glitch, deformed, mutated, ugly, disfigured",
+                )
                 num_steps = gr.Slider(
                     label="Number of steps",
                     minimum=1,
@@ -169,14 +187,15 @@ with gr.Blocks(css="style.css") as demo:
                     value=5,
                 )
                 adapter_conditioning_scale = gr.Slider(
-                    label="Adapter Conditioning Scale",
+                    label="Adapter conditioning scale",
                     minimum=0.5,
                     maximum=1,
                     step=0.1,
                     value=0.8,
                 )
-                cond_tau = gr.Slider(
-                    label="Fraction of timesteps for which adapter should be applied",
+                adapter_conditioning_factor = gr.Slider(
+                    label="Adapter conditioning factor",
+                    info="Fraction of timesteps for which adapter should be applied",
                     minimum=0.5,
                     maximum=1,
                     step=0.1,
@@ -201,7 +220,7 @@ with gr.Blocks(css="style.css") as demo:
         num_steps,
         guidance_scale,
         adapter_conditioning_scale,
-        cond_tau,
+        adapter_conditioning_factor,
         seed,
     ]
     prompt.submit(
@@ -238,7 +257,7 @@ with gr.Blocks(css="style.css") as demo:
         fn=run,
         inputs=inputs,
         outputs=result,
-        api_name="run",
+        api_name=False,
     )
 
 if __name__ == "__main__":

requirements.txt

@@ -1,8 +1,9 @@
 accelerate==0.22.0
 git+https://github.com/huggingface/diffusers@t2iadapterxl
-gradio==3.42.0
+gradio==3.43.1
 Pillow==10.0.0
 safetensors==0.3.3
 torch==2.0.1
 torchvision==0.15.2
-transformers==4.33.0
+transformers==4.33.1
+xformers==0.0.20