Update app.py

app.py

@@ -22,7 +22,7 @@ def generate(prompt, negative_prompt, model=image_model):
         pipe = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, token=hf_token).to('cuda')
         
         # Load the uncensored LoRA weights
-        pipe.load_lora_weights('enhanceaiteam/Flux-uncensored', weight_name='lora.safetensors')
+        pipe.load_lora_weights('enhanceaiteam/Flux-uncensored-v2', weight_name='lora.safetensors')
 
     else:
         pipe = DiffusionPipeline.from_pretrained(