File size: 632 Bytes
c05386f 8630717 c4e547f 8630717 3d654b9 67655f4 3d654b9 67655f4 4197898 3d654b9 2c15514 3d654b9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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() |