| from transformers import pipeline | |
| import gradio as gr | |
| sentiment = pipeline('sentiment-analysis') | |
| en_to_fr = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr") | |
| img_to_text = pipeline(model="ydshieh/vit-gpt2-coco-en") | |
| with gr.Blocks() as demo: | |
| with gr.Tab("Sentiment"): | |
| txt = gr.Textbox(label="Type your review", lines=2, | |
| placeholder="e.g. good place with great food") | |
| out1 = gr.Label() | |
| txt.submit(lambda t: sentiment(t)[0], inputs=txt, outputs=out1) | |
| with gr.Tab("Translation"): | |
| txt2 = gr.Textbox(label="Type text to translate", lines=2) | |
| out2 = gr.Textbox() | |
| txt2.submit(lambda t: en_to_fr(t)[0]['translation_text'], inputs=txt2, outputs=out2) | |
| with gr.Tab("Image → Text"): | |
| img = gr.Image(type="pil") | |
| out3 = gr.Textbox() | |
| img.change(lambda i: img_to_text(i)[0]['generated_text'], img, out3) | |
| demo.launch() | |