| import gradio as gr | |
| from huggingface_hub import InferenceClient | |
| client = InferenceClient("stabilityai/stable-diffusion-xl-base-1.0") | |
| clientTranslate = InferenceClient("Helsinki-NLP/opus-mt-vi-en") | |
| def translate(text): | |
| res = clientTranslate.translation(text) | |
| return res.translation_text | |
| def greet(prompt): | |
| text = translate(prompt) | |
| image = client.text_to_image(text) | |
| print(image) | |
| return image | |
| demo = gr.Interface( | |
| fn=greet, | |
| inputs=[ | |
| gr.Textbox(label="prompt") | |
| ], | |
| outputs=[gr.Image()], | |
| api_name="text-to-image", | |
| clear_btn=None | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() |