Jonny001's picture
Update app.py
f5bddfa verified
raw
history blame
881 Bytes
import gradio as gr
model1 = gr.load("models/pimpilikipilapi1/NSFW_master")
model2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
model3 = gr.load("models/prashanth970/flux-lora-uncensored")
def generate_images(text):
result_image1 = model1(text)
result_image2 = model2(text)
result_image3 = model3(text)
return result_image1, result_image2, result_image3
interface = gr.Interface(
fn=generate_images,
inputs=[
gr.Textbox(label="Type here your imagination:", placeholder="Type or click an example..."),
],
outputs=[
gr.Image(label="Model 1 Output"),
gr.Image(label="Model 2 Output"),
gr.Image(label="Model 3 Output")
],
description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
)
interface.launch()