Update app.py

app.py

@@ -16,8 +16,12 @@ model.eval()
 
 
 def translate_gradio(input):
-    encode = model.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt'))
-    text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]
+
+    with tokenizer.as_target_tokenizer():
+        encode = model.generate(tokenizer(text, return_tensors='pt'))
+    
+#    encode = model.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt'))
+        text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]
     return text_ar