import gradio as gr
from huggingface_hub import InferenceClient

client = InferenceClient("cutiee82/artsytistic-diffusion")

def infer(prompt):
    return client.text_to_image(prompt)

app = gr.Interface(
    infer,
    inputs=gr.Textbox(label="Prompt"),
    outputs=gr.Image(label="Result"),
    title="Artsytistic Diffusion"
)

if __name__ == "__main__":
    app.queue().launch(debug=True)