Mr-Vicky-01 commited on
Commit
2d6f21e
1 Parent(s): 00cae43

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -0
README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - ta
5
+ ---
6
+ ```python
7
+ def language_translator(text):
8
+ tokenizer = AutoTokenizer.from_pretrained(checkpoint)
9
+ model = AutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
10
+ # model = AutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
11
+ tokenized = tokenizer([text], return_tensors='pt')
12
+ out = model.generate(**tokenized, max_length=128)
13
+ with tokenizer.as_target_tokenizer():
14
+ return tokenizer.decode(out[0],skip_special_tokens=True)
15
+
16
+ text_to_translate = "How are you?"
17
+ output = language_translator(text_to_translate)
18
+ print(output)
19
+ '''