Spaces:
Runtime error
Runtime error
import gradio as gr # UI library | |
from transformers import pipeline # Transformers pipeline | |
model_checkpoint_en2fil = "SalamaThanks/SalamaThanksTransformer_en2fil_v2" | |
model_checkpoint_fil2en = "SalamaThanks/SalamaThanksTransformer_fil2en_v2" | |
translator_en2fil = pipeline("translation", model = model_checkpoint_en2fil) | |
translator_fil2en = pipeline("translation", model = model_checkpoint_fil2en) | |
def transformer_en2fil(from_text): | |
results = translator_en2fil(from_text) | |
return results[0]['translation_text'] | |
def transformer_fil2en(from_text): | |
results = translator_fil2en(from_text) | |
return results[0]['translation_text'] | |
def check_lang(lang, from_text): | |
if lang == "English-to-Filipino": | |
return transformer_en2fil(from_text) | |
elif lang == "Filipino-to-English": | |
return transformer_fil2en(from_text) | |
interface = gr.Interface( | |
fn=check_lang, | |
inputs=[ | |
gr.inputs.Radio(["English-to-Filipino", "Filipino-to-English"]), | |
gr.inputs.Textbox(lines=4, placeholder='Input Text to Translate:')], | |
outputs='text' | |
) | |
interface.launch(debug=True) |