Update app.py

app.py

@@ -36,19 +36,23 @@ controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-1
 
 repo_default = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
-repo_default.load_lora_weights("Corcelio/mobius", weight_name="mobius.safetensors", adapter_name="base2")
-repo_default.set_adapters(["base", "base2"], adapter_weights=[0.7, 0.7])
+repo_default.set_adapters(["base"], adapter_weights=[0.7])
 
 repo_pixel = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
 repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
 repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
 
+repo_large = StableDiffusionXLPipeline.from_pretrained("Corcelio/mobius", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
+repo_large.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
+repo_large.set_adapters(["base"], adapter_weights=[0.7])
+
 repo_customs = {
     "Default": repo_default,
     "Realistic": StableDiffusionXLPipeline.from_pretrained("stablediffusionapi/NightVision_XL", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=False, add_watermarker=False),
     "Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Pixel": repo_pixel,
+    "Large": repo_large,
 }
 
 # Functions
@@ -122,7 +126,7 @@ with gr.Blocks(css=css) as main:
         input = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Input")
         filter_input = gr.Textbox(lines=1, value="", label="Input Filter")
         negative_input = gr.Textbox(lines=1, value=DEFAULT_NEGATIVE_INPUT, label="Input Negative")
-        model = gr.Dropdown(label="Models", choices=["Default", "Realistic", "Anime", "Pixel"], value="Default")
+        model = gr.Dropdown(label="Models", choices=repo_customs.keys(), value="Default")
         height = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_HEIGHT, label="Height")
         width = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_WIDTH, label="Width")
         steps = gr.Slider(minimum=-1, maximum=100, step=1, value=-1, label="Steps")