Update app.py

app.py

@@ -24,14 +24,11 @@ DEFAULT_HEIGHT = 1024
 DEFAULT_WIDTH = 1024
 
 REPO = "sd-community/sdxl-flash"
-REPO_WEIGHT = "ehristoforu/dalle-3-xl-v2"
-WEIGHT = "dalle-3-xl-lora-v2.safetensors"
-ADAPTER = "dalle"
 
 model = StableDiffusionXLPipeline.from_pretrained(REPO, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 model.scheduler = EulerAncestralDiscreteScheduler.from_config(model.scheduler.config)
-model.load_lora_weights(REPO_WEIGHT, weight_name=WEIGHT, adapter_name=ADAPTER)
-model.set_adapters(ADAPTER, adapter_weights=[0.7])
+model.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="base")
+model.set_adapters(["base"], adapter_weights=[0.7])
 model.to(DEVICE)
 
 css = '''