jaksani commited on
Commit
a345371
·
verified ·
1 Parent(s): ebf2a19

Update README.md

Browse files

from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
model_checkpoint = "jaksani/english-to-telugu"

tokenizer = MBart50TokenizerFast.from_pretrained(model_checkpoint)
model = MBartForConditionalGeneration.from_pretrained(model_checkpoint)

text = "This is a test case." # Sentence to translate
inputs = tokenizer(text, return_tensors="pt")

outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0], skip_special_tokens=True)) # ఈ ఒక పరీక్ష కేసు ఉంది.

Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -4,4 +4,6 @@ language:
4
  - te
5
  - en
6
  pipeline_tag: translation
 
 
7
  ---
 
4
  - te
5
  - en
6
  pipeline_tag: translation
7
+ base_model:
8
+ - facebook/mbart-large-cc25
9
  ---