Mohamed Aymane Farhi
Add other models
9c8ab75
raw
history blame contribute delete
694 Bytes
import gradio as gr
from transformers import pipeline
models = [
"Helsinki-NLP/opus-mt-en-ber",
"Helsinki-NLP/opus-mt-ber-en",
"Helsinki-NLP/opus-mt-fr-ber",
"Helsinki-NLP/opus-mt-ber-fr",
"Helsinki-NLP/opus-mt-es-ber",
"Helsinki-NLP/opus-mt-ber-es",
"Helsinki-NLP/opus-mt-kab-en"
]
pipes = {}
def predict(text, model):
if model not in pipes:
pipes[model] = pipeline("translation", model=model)
pipe = pipes[model]
return pipe(text)[0]['translation_text']
demo = gr.Interface(
fn=predict,
inputs=[
gr.Textbox(lines=5, label="Input Text"),
gr.Dropdown(models, label="Model")
],
outputs='text',
)
demo.launch()