transformers_js_py
from transformers_js import import_transformers_js
import gradio as gr
transformers = await import_transformers_js()
pipeline = transformers.pipeline
pipe = await pipeline('translation', 'Xenova/nllb-200-distilled-600M')
async def generate(text):
return await pipe(text,
src_lang = "eng_Latn",
tgt_lang = "fra_Latn",)
demo = gr.Interface(
generate,
gr.Textbox(label="English Text"),
gr.JSON(label="Translated Text", flag=False),
examples=["My name is VB and I live in Stuttgart.", "I like eating Maultaschen and Wurst.", "Can I have a pain au chocolate, please?"],
theme=gr.themes.Soft(),)
demo.launch()