Create app.py

app.py

@@ -0,0 +1,134 @@
+#pip install gradio happytransformer
+
+import gradio as gr
+from transformers import pipeline
+from happytransformer import HappyTextToText, TTSettings
+
+def translate(language_1, language_2, message):
+  translator = pipeline(f"translation", model="Helsinki-NLP/opus-mt-{}-{}".format(language_1, language_2))
+  output = translator(message)
+  return output[0]['translation_text']
+
+def summarize(text):
+  summerizer = pipeline("summarization", model="facebook/bart-large-cnn")
+  output = summerizer(text)
+  return output[0]['summary_text']
+
+def question(question, context):
+  question_model = pipeline("question-answering", model="deepset/roberta-base-squad2")
+  output = question_model({
+      'question': question,
+      'context' : context
+  })
+  return output['answer']
+
+
+def grammar(user_input):
+    grammar_model = HappyTextToText("T5", "vennify/t5-base-grammar-correction")
+    args = TTSettings(num_beams=5, min_length=1)
+    output = grammar_model.generate_text('grammar: ' + user_input, args=args)
+    return output.text
+
+
+def mask(text):
+  mask_model = pipeline("fill-mask", model="google-bert/bert-base-uncased")
+  output = mask_model(text)
+  return output[0]['sequence']
+
+def multi_model(text):
+
+  mask_model = mask(text)
+
+  summerize_model = summarize(mask_model)
+
+  question_model = question('Is this text about politics?', summerize_model)
+
+  translation_model = translate('en', 'nl', question_model)
+
+  grammar_model = grammar(translation_model)
+
+  return grammar_model
+
+
+
+with gr.Blocks() as demo:
+  gr.Markdown("Technology 2 - Ai interfaces")
+
+#tab single model
+  with gr.Tab("Single-Models"):
+    with gr.Row(equal_height=True):
+
+      #column left (1)
+      with gr.Column(scale=1):
+
+        choice1 = gr.Dropdown(
+          choices=["nl", "fr", "en"],
+          label="Select an option")
+
+        choice2 = gr.Dropdown(
+          choices=["nl", "fr", "en"],
+          label="Select an option")
+
+        textbox = gr.Textbox(label="message")
+
+        translate_btn = gr.Button("Translate")
+
+        translated_box = gr.Textbox()
+
+        translate_btn.click(fn=translate,
+                            inputs=[choice1, choice2, textbox],
+                            outputs=translated_box )
+
+      #column right (2)
+      with gr.Column(scale=1):
+        textbox_summary = gr.Textbox(label="text you want to summarize")
+        textbox_summary_output = gr.Textbox()
+        summary_btn = gr.Button('summarize')
+
+        summary_btn.click(fn=summarize,
+                          inputs=textbox_summary,
+                          outputs=textbox_summary_output)
+
+      with gr.Column(scale=1):
+        textbox_question = gr.Textbox(label = 'question')
+        textbox_context = gr.Textbox(label = 'context')
+        textbox_question_output = gr.Textbox()
+        question_btn = gr.Button("Get your Answer")
+
+        question_btn.click(fn=question,
+                  inputs=[textbox_question, textbox_context],
+                  outputs = textbox_question_output)
+
+      with gr.Column(scale=1):
+        textbox_mask = gr.Textbox(label="Give me a sentance and use the word [MASK] to predict what will go there")
+        mask_button = gr.Button("Generate")
+        textbox_mask_output = gr.Textbox()
+
+        mask_button.click(fn=mask,
+                          inputs = textbox_mask,
+                          outputs = textbox_mask_output)
+
+      with gr.Column(scale=1):
+        textbox_grammar = gr.Textbox(label="I will correct your grammar mistakes.")
+        grammar_button = gr.Button("Fix grammar")
+        textbox_grammar_output = gr.Textbox()
+
+        grammar_button.click(fn=grammar,
+                             inputs=textbox_grammar,
+                             outputs=textbox_grammar_output)
+
+
+
+
+
+
+#tab mulit model
+  with gr.Tab("Multi-Models"):
+    textblock = gr.Textbox(label="give me a sentence and replace some words with [MASK] and i will summerize and translate it for your")
+    btn1 = gr.Button("START")
+    response = gr.Textbox()
+
+    btn1.click(fn=multi_model, inputs = textblock, outputs=response)
+
+
+demo.launch(debug=True)