| import gradio as gr | |
| model_1 = gr.load("models/pimpilikipilapi1/NSFW_master") | |
| model_2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux") | |
| model_3 = gr.load("models/prashanth970/flux-lora-uncensored") | |
| default_negative_prompt = ( | |
| "Extra limbs, Extra fingers or toes, Disfigured face, Distorted hands, Mutated body parts, " | |
| "Missing limbs, Asymmetrical features, Blurry face, Poor anatomy, Incorrect proportions, Crooked eyes, " | |
| "Deformed hands or fingers, Double face, Unrealistic skin texture, Overly smooth skin, Poor lighting, " | |
| "Cartoonish appearance, Plastic look, Grainy, Unnatural expressions, Crossed eyes, Mutated clothing, " | |
| "Artifacts, Uncanny valley, Doll-like features, Bad symmetry, Uneven skin tones, Extra teeth, " | |
| "Unrealistic hair texture, Dark shadows on face, Overexposed face, Cluttered background, Text, watermark, " | |
| "or signature, Over-processed, Low quality, Blurry, Low resolution, Pixelated, Oversaturated, Too dark, Overexposed, Poor lighting, " | |
| "Unclear, Text or watermark, Distorted faces, Extra limbs or fingers, Disfigured, Grainy, Overly stylized, Cartoonish, " | |
| "Unrealistic anatomy, Bad proportions, Unrealistic colors, Artificial look, Background noise, Unwanted objects, Repetitive patterns, Artifacting, Abstract shapes, Out of focus" | |
| ) | |
| def generate_image(model, prompt, negative_prompt): | |
| """Generate image using the given model.""" | |
| prompt += " high-resolution" | |
| try: | |
| output = model(prompt, negative_prompt=negative_prompt) | |
| except TypeError: | |
| output = model(prompt) | |
| if isinstance(output, tuple) and len(output) > 0: | |
| output = output[0] | |
| return output | |
| interface = gr.Interface( | |
| fn=lambda prompt, negative_prompt: ( | |
| generate_image(model_1, prompt, negative_prompt), | |
| generate_image(model_2, prompt, negative_prompt), | |
| generate_image(model_3, prompt, negative_prompt), | |
| ), | |
| inputs=[ | |
| gr.Textbox(label="Type your prompt here: ✍️", placeholder="Describe what you want..."), | |
| gr.Textbox(label="Negative prompt:", value=default_negative_prompt), | |
| ], | |
| outputs=[ | |
| gr.Image(label="Generated Image - Model 1"), | |
| gr.Image(label="Generated Image - Model 2"), | |
| gr.Image(label="Generated Image - Model 3"), | |
| ], | |
| title="Text to Image (NSFW) 🔞", | |
| description="⚠️ Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.", | |
| ) | |
| interface.launch() |