import gradio as gr
from diffusers import DiffusionPipeline
import torch

print("Lade Modell...")
model_id = "votepurchase/pornmasterPro_realismILV4"
pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

def generate(prompt):
    return pipe(prompt).images[0]

gr.Interface(
    fn=generate,
    inputs=gr.Textbox(label="Prompt"),
    outputs=gr.Image(label="Bild"),
    title="Pornmaster Demo"
).launch()