File size: 860 Bytes
c05386f 8630717 9df9ab1 c4e547f 8630717 3d654b9 67655f4 9df9ab1 67655f4 9df9ab1 e07d862 af8c607 67655f4 3d654b9 9845ae1 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 30 31 32 33 34 35 36 37 |
import gradio as gr
from huggingface_hub import InferenceClient
from transformers import pipeline
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 translate(text):
model = f"Helsinki-NLP/opus-mt-vi-en"
pipe = pipeline("translation", model=model)
translation = pipe(text)
return translation[0]['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() |