Update app.py

app.py

@@ -1,12 +1,11 @@
-from transformers import AutoTokenizer, TFAutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import tensorflow as tf
 import gradio as gr
 checkpoint = "Hemanth-thunder/english-tamil-mt"
 
 def language_translator(text):
     tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-    model = TFAutoModelForSeq2SeqLM.from_pretrained("finetune-EN-to-Ta/")
-    # model = TFAutoModelForSeq2SeqLM.from_pretrained("checkpoint")
+    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():