Update app.py

app.py

@@ -15,11 +15,8 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 
 # pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
 
-lora_repo = "Shakker-Labs/AWPortraitCN"
-trigger_word = ""  # Leave trigger_word blank if not used.
-pipe.load_lora_weights(lora_repo, low_cpu_mem_usage=True)
-
-pipe.to("cuda")
+pipe.load_lora_weights("Shakker-Labs/AWPortraitCN2", weight_name="AWPortraitCN_2.safetensors")
+pipe.fuse_lora()
 
 MAX_SEED = 2**32-1