Update app.py

app.py

@@ -1,17 +1,12 @@
-import gradio as gr
+from diffusers import AutoPipelineForText2Image
+import torch
 
-# Load the base model using the Hugging Face Inference API
-def model_interface():
-    return gr.load("black-forest-labs/FLUX.1-dev", src="models")
+# Load the base model
+pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
 
-# Load the LoRA
-def lora_interface():
-    return gr.load("models/enhanceaiteam/Flux-Uncensored-V2", lora=True)
+# Load the uncensored LoRA weights
+pipeline.load_lora_weights('enhanceaiteam/Flux-uncensored-v2', weight_name='lora.safetensors')
 
-# Initialize the interfaces
-base_model = model_interface()
-lora_model = lora_interface()
-
-# Launch the interfaces
-base_model.launch()
-lora_model.launch()
+# Generate an image with an uncensored NSFW prompt
+image = pipeline('a naked cute girl').images[0]
+image.show()