Jonny001's picture
Update app.py
f94ea70 verified
raw
history blame
1.18 kB
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()