Update app.py

app.py

@@ -14,7 +14,7 @@ if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
-CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
+CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
 USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
@@ -57,7 +57,7 @@ def generate(
     if style=="BEST" : 
         pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle2")
         pipe.load_lora_weights("ehristoforu/dalle-3-xl", weight_name="dalle-3-xl-lora-v1.safetensors", adapter_name="dalle1")
-        pipe.set_adapters(["dalle2","dalle1"], adapter_weights=[1.5, 0.5])
+        pipe.set_adapters(["dalle2","dalle1"], adapter_weights=[0.7, 0.3])
     elif style=="Origami":
         pipe.load_lora_weights("RalFinger/origami-style-sdxl-lora", weight_name="ral-orgmi-sdxl.safetensors", adapter_name="origami")
         pipe.set_adapters(["origami"], adapter_weights=[2])
@@ -74,15 +74,15 @@ def generate(
         pipe.load_lora_weights("artificialguybr/LogoRedmond-LogoLoraForSDXL", weight_name="LogoRedmond_LogoRedAF.safetensors", adapter_name="pixel")
         pipe.set_adapters(["lora", "pixel"], adapter_weights=[0.5, 1.2])
     else: 
-        pipe.load_lora_weights("artificialguybr/StickersRedmond", weight_name="StickersRedmond.safetensors", adapter_name="lora")
-        pipe.set_adapters(["orignal"], adapter_weights=[0.1])
-
+        pipe.load_lora_weights()
+        
     pipe.to("cuda")
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)   
 
     options = {
         "prompt":prompt,
+        "style":style,
         "negative_prompt":negative_prompt,
         "width":width,
         "height":height,