|
--- |
|
license: apache-2.0 |
|
widget: |
|
- text: >- |
|
I was born at 2003 and my sister was born at 2000. |
|
- text: >- |
|
How are You? |
|
- text: >- |
|
Actor Vijay is competing an 2026 election. |
|
language: |
|
- ta |
|
--- |
|
```python |
|
def language_translator(text): |
|
tokenizer = AutoTokenizer.from_pretrained(checkpoint) |
|
model = AutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/") |
|
# model = AutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/") |
|
tokenized = tokenizer([text], return_tensors='pt') |
|
out = model.generate(**tokenized, max_length=128) |
|
with tokenizer.as_target_tokenizer(): |
|
return tokenizer.decode(out[0],skip_special_tokens=True) |
|
|
|
text_to_translate = "How are you?" |
|
output = language_translator(text_to_translate) |
|
print(output) |
|
``` |