|
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") |
|
|
|
def generate_image_model_1(prompt): |
|
prompt += " 2k" |
|
return model_1(prompt) |
|
|
|
def generate_image_model_2(prompt): |
|
prompt += " 8k" |
|
return model_2(prompt) |
|
|
|
def generate_image_model_3(prompt): |
|
prompt += " 10k" |
|
return model_3(prompt) |
|
|
|
interface = gr.Interface( |
|
fn=lambda prompt: ( |
|
generate_image_model_1(prompt), |
|
generate_image_model_2(prompt), |
|
generate_image_model_3(prompt) |
|
), |
|
inputs=[ |
|
gr.Textbox(label="Type your prompt here: โ๏ธ", placeholder="Describe what you want..."), |
|
], |
|
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) ๐", |
|
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() |