Mr-Vicky-01 commited on
Commit
68ec033
·
verified ·
1 Parent(s): 33a6a47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ checkpoint = "Hemanth-thunder/english-tamil-mt"
4
 
5
  def language_translator(text):
6
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
7
- # model = TFAutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
8
- model = TFAutoModelForSeq2SeqLM.from_pretrained("checkpoint")
9
  tokenized = tokenizer([text], return_tensors='np')
10
  out = model.generate(**tokenized, max_length=128)
11
  with tokenizer.as_target_tokenizer():
 
4
 
5
  def language_translator(text):
6
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
7
+ model = TFAutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
8
+ # model = TFAutoModelForSeq2SeqLM.from_pretrained("checkpoint")
9
  tokenized = tokenizer([text], return_tensors='np')
10
  out = model.generate(**tokenized, max_length=128)
11
  with tokenizer.as_target_tokenizer():