Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 import uuid
 import os
 
-from diffusers import StableDiffusionXLPipeline
+from diffusers import StableDiffusionXLPipeline, StableDiffusion3Pipeline
 from transformers import pipeline
 from PIL import Image
 
@@ -43,11 +43,16 @@ repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="ba
 repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
 repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
 
+repo_large = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", 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("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Pixel": repo_pixel,
+    "Large": repo_large,
 }
 
 # Functions
@@ -84,8 +89,11 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
     elif model == "Pixel":   
         steps_set = 15
         guidance_set = 1.5
+    elif model == "Large":   
+        steps_set = 25
+        guidance_set = 5
     else:
-        steps_set = 20
+        steps_set = 25
         guidance_set = 7
 
     if not steps: