Update app.py

app.py

@@ -13,6 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
 good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=dtype).to(device)
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, vae=taef1).to(device)
+pipe.load_lora_weights("enhanceaiteam/Flux-Uncensored-V2")
 torch.cuda.empty_cache()
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -23,7 +24,7 @@ pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_ret
 @spaces.GPU(duration=25)
 def infer(prompt, seed=42, randomize_seed=True, width=768, height=768, guidance_scale=4.5, num_inference_steps=24, progress=gr.Progress(track_tqdm=True)):
     """
-    Generate natural, realistic professional-quality images using the Flux.1 Krea-Dev Image Generator, eliminating typical AI image artifacts. Specialized for Product photography, e-commerce visuals, concept art, fashion and beauty photography, stock images
+    Generate natural, realistic professional-quality images using the Flux.1 dev base model paired with the Flux-Uncensored-V2 LoRA to remove base model content restrictions. Specialized for product photography, e-commerce visuals, concept art, fashion and beauty photography, and stock images
 
     Args:
        prompt (str): Approximately 60-70 words max. Detailed description emphasizing technical photography terms, specific lighting, and material textures. Include professional camera/lens specifications for photorealistic results.
@@ -61,9 +62,9 @@ css="""
 with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# FLUX.1 Krea [dev]
-FLUX.1 Krea [dev] model further tuned and customized with [Krea](https://krea.ai)
-[[non-commercial license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)] [[blog](https://blackforestlabs.ai/announcing-black-forest-labs/)] [[model](https://huggingface.co/black-forest-labs/FLUX.1-dev)]
+        gr.Markdown(f"""# Flux-Uncensored-V2 (LoRA)
+    Flux.1 dev base model enhanced with the Flux-Uncensored-V2 LoRA for unrestricted generation.
+    [[license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md)] [[blog](https://blackforestlabs.ai/announcing-black-forest-labs/)] [[base model](https://huggingface.co/black-forest-labs/FLUX.1-dev)] [[LoRA](https://huggingface.co/enhanceaiteam/Flux-Uncensored-V2)]
         """)
         
         with gr.Row():