Create mt0_translate_eng_darija

mt0_translate_eng_darija

@@ -0,0 +1,41 @@
+import os
+import torch
+import gradio as gr
+import time
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+
+def translation(text):
+
+    model_checkpoint = "bigscience/mt0-base"
+    tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to('cuda')
+    #inference
+    inputs = tokenizer("translate to arabic : hi, I am salma and you?", return_tensors="pt").to("cuda")
+    output = model.generate(**inputs)
+    output = tokenizer.decode(output.cpu().numpy()[0])
+
+    return output
+
+
+if __name__ == '__main__':
+    print('\tinit models')
+
+    #inputs = [gr.inputs.Radio(['nllb-distilled-600M', 'nllb-1.3B', 'nllb-distilled-1.3B'], label='NLLB Model'),
+    inputs = [gr.inputs.Textbox(lines=5, label="Input text")]
+
+    outputs = gr.outputs.Textbox(label="Input text")
+
+    title = "NLP Translation model from english to darija"
+
+    demo_status = "Demo is running on CPU"
+    description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
+    examples = [
+    ['English', 'Darija', 'Hi nice to meet you']
+    ]
+
+    gr.Interface(translation,
+                 inputs,
+                 outputs,
+                 title=title,
+                 description=description,
+                 ).launch(debug=True)