Update app.py

app.py

@@ -18,6 +18,11 @@ pipe.to("cuda")
 refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
 refiner.to("cuda")
 
+pipe_fast = StableDiffusionXLPipeline.from_pretrained( "SG161222/RealVisXL_V4.0_Lightning", torch_dtype=torch.float16, vae=vae)
+pipe_fast.load_lora_weights("KingNish/Better-Image-XL-Lora", weight_name="example-03.safetensors", adapter_name="lora")
+pipe_fast.set_adapters("lora")
+pipe_fast.to("cuda")
+
 help_text = """
 To optimize image results:
 - Adjust the **Image CFG weight** if the image isn't changing enough or is changing too much. Lower it to allow bigger changes, or raise it to preserve original details.
@@ -55,17 +60,18 @@ def king(type ,
         instruction: str ,
         steps: int = 25,
         randomize_seed: bool = False,
-        seed: int = 25,
+        seed: int = 2404,
         width: int = 1024,
         height: int = 1024,
         guidance_scale: float = 7,
+        fast=True,
         use_resolution_binning: bool = True,
         progress=gr.Progress(track_tqdm=True),
     ):
     if type=="Image Editing" :
         raw_image = Image.open(input_image).convert('RGB')
         if randomize_seed:
-            seed = random.randint(0, 99999)
+            seed = random.randint(0, 999999)
         generator = torch.manual_seed(seed)
         output_image = pipe_edit(
             instruction, image=raw_image,
@@ -82,24 +88,23 @@ def king(type ,
         return seed, refine
     else :
         if randomize_seed:
-            seed = random.randint(0, 99999)
+            seed = random.randint(0, 999999)
         generator = torch.Generator().manual_seed(seed)
-        image = pipe(
-            prompt = instruction,
+        if fast:
+            pipes=pipe_fast
+        else:
+            pipes=pipe
+        image = pipes( prompt = instruction,
             guidance_scale = guidance_scale, 
             num_inference_steps = steps, 
-            width = (width), 
-            height = (height),
-            generator = generator,
-            output_type="latent",
+            width = width, height = height,
+            generator = generator, output_type="latent",
         ).images         
 
-        refine = refiner(
-            prompt=instruction,
+        refine = refiner( prompt=instruction,
             guidance_scale=guidance_scale,
             num_inference_steps=steps,
-            image=image,
-            generator=generator,
+            image=image, generator=generator,
         ).images[0]        
         return seed, refine
 
@@ -165,37 +170,49 @@ examples=[
         ],
     ]
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     gr.Markdown("# Image Generator Pro")
     with gr.Row():
-        instruction = gr.Textbox(lines=1, label="Instruction", interactive=True)
-    with gr.Row():
-        with gr.Column(scale=1):
-            type = gr.Dropdown(["Image Generation","Image Editing"], label="Task", value="Image Generation",interactive=True)
-            enhance_prompt = gr.Checkbox(label="Enhance prompt", value = True)
+        with gr.Column(scale=2):
+            instruction = gr.Textbox(lines=1, label="Instruction", interactive=True)
         with gr.Column(scale=1):
             generate_button = gr.Button("Generate")
+    with gr.Row():
+        type = gr.Dropdown(["Image Generation","Image Editing"], label="Task", value="Image Generation",interactive=True)
+        enhance_prompt = gr.Checkbox(label="Enhance prompt", value = True)
+        fast = gr.Checkbox(label="FAST Generation")
 
     with gr.Row():
         input_image = gr.Image(label="Image", type='filepath', interactive=True)
 
     with gr.Row():
-        guidance_scale = gr.Number(value=6.0, step=0.1, label="Image Generation Guidance Scale", interactive=True)
+        guidance_scale = gr.Number(value=6.0, step=0.1, label="Guidance Scale", interactive=True)
         steps = gr.Number(value=25, step=1, label="Steps", interactive=True)
 
-    with gr.Row():
-        width =  gr.Slider( label="Width", minimum=256, maximum=2048, step=64, value=1024)
-        height =  gr.Slider( label="Height", minimum=256, maximum=2048, step=64, value=1024)
-
-    with gr.Row():
-        randomize_seed = gr.Radio(
-                ["Fix Seed", "Randomize Seed"],
-                value="Randomize Seed",
-                type="index",
-                show_label=False,
-                interactive=True,
-            )
-        seed = gr.Number(value=1371, step=1, label="Seed", interactive=True)
+    with gr.Accordion("Advanced options", open=False):
+        with gr.Row():
+            with gr.Column(scale=1):
+                use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
+            with gr.Column(scale=2):
+                negative_prompt = gr.Text(
+                    label="Negative prompt",
+                    max_lines=1,
+                    placeholder="Enter a negative prompt",
+                    value="(deformed iris, deformed pupils, semi-realistic, cropped, out of frame, worst quality, low quality, jpeg artifacts, ugly, duplicate, extra fingers, poorly drawn hands, poorly drawn face, deformed, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck, (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, ugly, disgusting,",
+                    visible=True,
+                )
+        with gr.Row():
+            width =  gr.Slider( label="Width", minimum=256, maximum=2048, step=64, value=1024)
+            height =  gr.Slider( label="Height", minimum=256, maximum=2048, step=64, value=1024)
+        with gr.Row():
+            randomize_seed = gr.Radio(
+                    ["Fix Seed", "Randomize Seed"],
+                    value="Randomize Seed",
+                    type="index",
+                    show_label=False,
+                    interactive=True,
+                )
+            seed = gr.Number(value=1371, step=1, label="Seed", interactive=True)
 
 
 
@@ -227,6 +244,7 @@ with gr.Blocks(css=css) as demo:
                 width,
                 height,
                 guidance_scale,
+                fast,
             ],
             outputs=[seed, input_image],
         )