| import gradio as gr | |
| models = { | |
| "NSFW Master": gr.load("models/pimpilikipilapi1/NSFW_master"), | |
| "NSFW TrioHMH Flux": gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux"), | |
| "Flux LoRA Uncensored": 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(text, negative_prompt): | |
| result_images = {} | |
| for model_name, model in models.items(): | |
| try: | |
| result_images[model_name] = model(text, negative_prompt=negative_prompt) | |
| except TypeError: | |
| result_images[model_name] = model(text) | |
| return [result_images[model_name] for model_name in models] | |
| interface = gr.Interface( | |
| fn=generate_image, | |
| 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=model_name) for model_name in models], | |
| theme="NoCrypt/miku", | |
| description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.", | |
| ) | |
| interface.launch() |