Update app.py

app.py

@@ -24,6 +24,29 @@ pipe_best.load_lora_weights("KingNish/Better-Image-XL-Lora", weight_name="exampl
 pipe_best.set_adapters(["lora","dalle"], adapter_weights=[1.5, 0.7])
 pipe_best.to("cuda")
 
+pipe_ori = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae)
+pipe_ori.load_lora_weights("RalFinger/origami-style-sdxl-lora", weight_name="ral-orgmi-sdxl.safetensors", adapter_name="origami")
+pipe_ori.set_adapters(["origami"], adapter_weights=[2])
+pipe_ori.to("cuda")
+
+pipe_3D = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae)
+pipe_3D.load_lora_weights("artificialguybr/3DRedmond-V1", weight_name="3DRedmond-3DRenderStyle-3DRenderAF.safetensors", adapter_name="dalle2")
+pipe_3D.load_lora_weights("goofyai/3d_render_style_xl", weight_name="3d_render_style_xl.safetensors", adapter_name="dalle1")
+pipe_3D.set_adapters(["dalle2","dalle1"], adapter_weights=[1.1, 0.8])
+pipe_3D.to("cuda")
+
+pipe_pixel = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae)
+pipe_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", weight_name="PixelArtRedmond-Lite64.safetensors", adapter_name="lora")
+pipe_pixel.load_lora_weights("nerijs/pixel-art-xl", weight_name="pixel-art-xl.safetensors", adapter_name="pixel")
+pipe_pixel.set_adapters(["lora", "pixel"], adapter_weights=[1.0, 1.2])
+pipe_pixel.to("cuda")
+
+pipe_logo = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae)
+pipe_logo.load_lora_weights("artificialguybr/StickersRedmond", weight_name="StickersRedmond.safetensors", adapter_name="lora")
+pipe_logo.load_lora_weights("artificialguybr/LogoRedmond-LogoLoraForSDXL", weight_name="LogoRedmond_LogoRedAF.safetensors", adapter_name="pixel")
+pipe_logo.set_adapters(["lora", "pixel"], adapter_weights=[0.5, 1.2])
+pipe_logo.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.
@@ -68,7 +91,7 @@ def king(type ,
         image_cfg_scale: float = 1.7,
         width: int = 1024,
         height: int = 1024,
-        guidance_scale: float = 6,
+        style="BEST",
         use_resolution_binning: bool = True,
         progress=gr.Progress(track_tqdm=True),
     ):
@@ -90,7 +113,16 @@ def king(type ,
         if randomize_seed:
             seed = random.randint(0, 99999)
         generator = torch.Generator().manual_seed(seed)
-        image = pipe_best( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]         
+        if style=="3D"
+            image = pipe_3D( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]         
+        elif style=="PixelArt"
+            image = pipe_pixel( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]         
+        elif style=="Logo"
+            image = pipe_logo( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]         
+        elif style=="Origami"
+            image = pipe_ori( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]         
+        else:
+            image = pipe_best( prompt = instruction, guidance_scale = 5, num_inference_steps = steps, width = width, height = height, generator = generator).images[0]         
         return seed, image
 
 client = InferenceClient()
@@ -160,7 +192,8 @@ with gr.Blocks(css=css) as demo:
             type = gr.Dropdown(["Image Generation","Image Editing"], label="Task", value="Image Generation",interactive=True, info="AI will select option based on your query, but if it selects wrong, please choose correct one.")
         with gr.Column(scale=1):
             generate_button = gr.Button("Generate")
-
+    with gr.Row():
+        style = gr.Radio(choices=["BEST","3D", "PixelART","Logo","Origami"],label="Style", value="BEST", interactive=True)
     with gr.Row():
         input_image = gr.Image(label="Image", type="pil", interactive=True)
 
@@ -207,7 +240,8 @@ with gr.Blocks(css=css) as demo:
                 text_cfg_scale,
                 image_cfg_scale,
                 width,
-                height
+                height,
+                style
             ],
             outputs=[seed, input_image],
         )