application file added

app.py

@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+"""translator.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1qSV5usAOci7XzgKcLVYpYMok5RpFBYSQ
+
+# Text Translation
+"""
+
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+import gradio as gr
+from keras.utils.generic_utils import default
+
+#!pip install -U pip transformers
+#!pip install sentencepiece
+
+checkpoint = "facebook/nllb-200-distilled-600M"
+# checkpoint = ‘facebook/nllb-200–1.3B’
+# checkpoint = ‘facebook/nllb-200–3.3B’
+# checkpoint = ‘facebook/nllb-200-distilled-1.3B’
+
+model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+
+def translate(source, target, text):
+    formats = {'English':'eng_Latn', 'Asante': 'aka_Latn', 'Akuapem': 'twi_Latn', 'Ewe':'ewe_Latn', 'Hausa': 'hau_Latn'}
+    source_fmt = formats[source]
+    target_fmt = formats[target]
+    translator = pipeline('translation', 
+                      model=model, 
+                      tokenizer=tokenizer, 
+                      src_lang=source_fmt, 
+                      tgt_lang=target_fmt, 
+                      max_length = 400)
+    
+    output = translator(text)
+    translated_text = output[0]['translation_text']
+    return translated_text
+
+title = "Ananse AI | Ghanaian Language Translator"
+
+demo = gr.Interface(
+    fn=translate,
+    inputs=[gr.inputs.Dropdown(choices=['Akuapem','Asante', 'English', 'Ewe','Hausa'], label='Source Language'), gr.inputs.Dropdown(choices=['Akuapem','Asante', 'English', 'Ewe','Hausa'], label='Target Language'), gr.inputs.Textbox(lines=5, label='Text to Translate')],
+    outputs=[gr.outputs.Textbox(label='Translated Text')],
+    title=title,
+    #description = 'This is a translator to translate popular Ghanaian languages.',
+    #article='Ananse AI | hnmensah',
+    examples = [['English','Asante','Kame went to Kaneshie to buy tomates.'],
+                ['English','Ewe','The event should be hosted at the Accra Mall.'],
+                ['English','Akuapem','The trader is suffering from Malaria so she did not go to work.'],
+                ['English','Hausa','The last person to get to the class will be sacked.']],
+)
+
+demo.launch(share=True, debug=False)