Update app.py

app.py

@@ -1,12 +1,18 @@
 import gradio as gr
 from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
+model = torch.load("helsinki_fineTuned.pt")
+translation_pipeline = pipeline(model)
 
-translation_pipeline = pipeline("translation_en_to_fr")
 
 def translate_gradio(input):
-    result = translation_pipeline(input)
-    return result[0]['translation_text']
+    encode = translation_pipeline.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt').to('cuda'))
+    text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]
+    return text_ar
+
+
 
 translate_interface = gr.Interface(fn = translate_gradio,
                                    inputs="text",