Update app.py

app.py

@@ -36,8 +36,13 @@ if not torch.cuda.is_available():
 
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
-lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
-trigger_word = "Ultra realistic"  # Leave trigger_word blank if not used.
+
+lora_repo = "prithivMLmods/Green-Cartoon-Flux-LoRA"
+trigger_word = "Green Cartoon"  # Leave trigger_word blank if not used.
+
+#lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
+#trigger_word = "Ultra realistic"  # Leave trigger_word blank if not used.
+
 pipe.load_lora_weights(lora_repo)
 pipe.to("cuda")