Update app.py

app.py

@@ -42,6 +42,9 @@ footer {
 repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
 
 repo_default = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
+repo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="default_base")
+repo.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
+repo.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
 
 # repo_large = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, add_watermarker=False, revision="refs/pr/1")
 # repo_large.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
@@ -51,7 +54,7 @@ repo_customs = {
     "Default": repo_default,
     # "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     # "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
-    # "Pixel": repo_default,
+    "Pixel": repo_default,
     # "Large": repo_neo,
 }
 
@@ -71,7 +74,7 @@ def get_seed(seed):
 @spaces.GPU(duration=60)
 def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATIVE_INPUT, model=DEFAULT_MODEL, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, number=1, seed=None):
 
-    repo = repo_customs[model or "Default"]
+    repo = repo_customs[model or "Default"].to(DEVICE)
     filter_input = filter_input or ""
     negative_input = negative_input or DEFAULT_NEGATIVE_INPUT
     steps_set = steps
@@ -89,22 +92,14 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
     elif model == "Pixel":   
         steps_set = 15
         guidance_set = 1.5
-        
-        repo.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
-        repo.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
-        repo.set_adapters(["base", "base2"], adapter_weights=[1, 1])
+        repo.set_adapters(["pixel_base", "pixel_base_2"], adapter_weights=[1, 1])
     elif model == "Large":
         steps_set = 25
         guidance_set = 3.5
     else:
         steps_set = 25
         guidance_set = 7
-        print("1")
-        repo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
-        print(2)
-        repo.set_adapters(["base"], adapter_weights=[0.7])
-        print(3)
-        repo.to(DEVICE)
+        repo.set_adapters(["default_base"], adapter_weights=[0.7])
 
     if not steps:
         steps = steps_set