Update app.py

app.py

@@ -24,8 +24,8 @@ def generate_image(image, prompt, guidance_scale, num_steps, lora_name):
     # Load and fuse base LoRA weights
     
     # Load selected LoRA effect if not using the pretrained base model
-    pipeline.load_lora_weights("nicolaus-huang/PhotoDoodle", weight_name=f"{lora_name}.safetensors")
-    pipeline.fuse_lora()
+    if args.lora_name != 'pretrained':
+        pipeline.load_lora_weights("nicolaus-huang/PhotoDoodle", weight_name=f"{lora_name}.safetensors")
     
     height=768
     width=512