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