Update app.py

app.py

@@ -1,42 +1,26 @@
 from transformers import pipeline
 import gradio as gr
 
-
 sentiment = pipeline('sentiment-analysis')
-
-def get_sentiment(text):
-  response = sentiment(text)
-
-  return response[0]['label'], response[0]['score']
-
-textbox = gr.Textbox(label="Type your review", placeholder="example: good place with great food", lines=2)
-
-gr.Interface(
-    get_sentiment, 
-    textbox, #'text', 
-    "text").launch()
-
-
-### TRanslation
 en_to_fr = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
-def translate(input):
-  return en_to_fr(input)
+img_to_text = pipeline(model="ydshieh/vit-gpt2-coco-en")
 
-textbox = gr.Textbox(label="Type your original text", placeholder="example: blah blah", lines=2)
+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)
 
-gr.Interface(
-    translate,
-    textbox, #'text',
-    "text").launch()
+    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)
 
-####
-img_to_text = pipeline(model="ydshieh/vit-gpt2-coco-en")
-def img_text(img):
-  return img_to_text(img)
+demo.launch()
 
-img = gr.Image()
-gr.Interface(
-    img_to_text,
-    img, #'text',
-    "text").launch()