Update app.py

app.py

@@ -16,16 +16,23 @@ model.eval()
 
 
 def translate_gradio(input):
-
+    '''
     with tokenizer.as_target_tokenizer():
         input_ids = tokenizer(input, return_tensors='pt')
     encode = model.generate(**input_ids)
 #    encode = model.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt'))
+    text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]'''
+    
+    tokenized_text = tokenizer.prepare_seq2seq_batch([input], return_tensors='pt')
+    
+    # Perform translation and decode the output
+    encode = model.generate(**tokenized_text)
     text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]
     return text_ar
 
 
 
+
 translate_interface = gr.Interface(fn = translate_gradio,
                                    inputs="text",
                                    outputs="text"                      )