import gradio as gr
from huggingface_hub import InferenceClient

client = InferenceClient("stabilityai/stable-diffusion-xl-base-1.0")

def greet(prompt):
    print(prompt)
    image = client.text_to_image(prompt)
    print(image)
    return image

demo = gr.Interface(
    fn=greet,
    inputs=[
        gr.Textbox(label="prompt")
    ],
    outputs=[gr.Image()],
    api_name="chat",
    clear_btn=None
)

if __name__ == "__main__":
    demo.launch()