Upload 2 files

app.py

@@ -0,0 +1,14 @@
+from transformers import AutoModelForSeq2SeqLM,AutoTokenizer
+import gradio as grad
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+model_c = 'Ahmed107/nllb200-ar-en'
+tokenizer = AutoTokenizer.from_pretrained(model_c,src_lang='ajp_Arab', tgt_lang='eng_Latn')
+model = AutoModelForSeq2SeqLM.from_pretrained(model_c)
+
+def translate(text):
+  inputs = tokenizer(text, return_tensors="pt")
+  outputs = model.generate(**inputs, max_length=128, num_beams=5, early_stopping=True)
+  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+  return response
+grad.Interface(translate, inputs=["text",], outputs="text").launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+transformers
+torch
+transformers[sentencepiece]