Update app.py

app.py

@@ -55,10 +55,12 @@ REPO_ID_DISTILLED = "black-forest-labs/FLUX.2-klein-9B"
 print("Loading 9B Regular model...")
 pipe_regular = Flux2KleinPipeline.from_pretrained(REPO_ID_REGULAR, torch_dtype=dtype)
 pipe_regular.to("cuda")
+pipe_regular.load_lora_weights(".", weight_name="Myctor_epoch_10.safetensors")
 
 print("Loading 9B Distilled model...")
 pipe_distilled = Flux2KleinPipeline.from_pretrained(REPO_ID_DISTILLED, torch_dtype=dtype)
 pipe_distilled.to("cuda")
+pipe_distilled.load_lora_weights(".", weight_name="Myctor_epoch_10.safetensors")
 
 # Dictionary for easy access
 pipes = {