| 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() |