import transformers
import gradio as gr
from transformers import pipeline
import os 
TOKEN = os.getenv('HUGGING_FACE_HUB_TOKEN')



translator = pipeline("translation", model="barghavani/en_to_fr_translation_model",use_auth_token=TOKEN)

tran_fun = lambda x: translator(x)[0]['translation_text']


demo = gr.Interface(
    fn = tran_fun, 
    inputs = gr.Textbox(label = 'Enter An English Word to be translated into French', lines = 3),
    outputs = 'text',
)

demo.launch(share=True,debug=True)