Update app.py

app.py

@@ -86,6 +86,9 @@ def run_lora(prompt, negative, weight, selected_state):
                 lora_model = lora_model.to("cuda")
                 lora_model.apply_to(pipe.text_encoder, pipe.unet)
                 lora_model = lora_model.to("cuda")
+                lora_model.merge_to(
+                    pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
+                )
             last_merged = True
 
     image = pipe(