Spaces:

Mr-Vicky-01
/

tamil_translator

Sleeping

Mr-Vicky-01 commited on Mar 12

Commit

12fdfea

•

1 Parent(s): 4dd5512

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as gr
 checkpoint = "Hemanth-thunder/english-tamil-mt"
 def language_translator(text):
-    tokenizer = AutoTokenizer.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)

 checkpoint = "Hemanth-thunder/english-tamil-mt"
 def language_translator(text):
+    tokenizer = AutoTokenizer.from_pretrained(checkpoint)
     model = AutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
     tokenized = tokenizer([text], return_tensors='pt')
     out = model.generate(**tokenized, max_length=128)