Update app.py

app.py

@@ -10,6 +10,10 @@ LORA_MODEL = "MegaTronX/MetartLoRA" # Replace with your repo
 # Load base model pipeline
 pipe = FluxPipeline.from_pretrained(BASE_MODEL, torch_dtype=torch.float16)
 
+#pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+#pipe.load_lora_weights("MegaTronX/MetartLoRA")
+
+
 # Load LoRA weights
 pipe.unet.load_attn_procs(LORA_MODEL)
 pipe = pipe.to("cuda")