File size: 2,519 Bytes
17e58be 77efc8c 96ad892 945386e 77efc8c 3a1878a c7b40e1 3a1878a c7b40e1 b4380ae 945386e b4380ae 945386e b4380ae 0862982 b4380ae 0862982 c7b40e1 0389d06 0aad51a b4380ae 0aad51a 0389d06 945386e 3a1878a 0389d06 945386e 0389d06 146f57d b4380ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
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() |