lingvanex's picture
Update app.py
02b79f6
raw
history blame contribute delete
601 Bytes
import gradio as gr
import sentencepiece as spm
import ctranslate2
description = "Translator"
title = "Translate!"
translator = ctranslate2.Translator("./", device="cpu")
spm_encoder = spm.SentencePieceProcessor("./en.spm.model")
spm_decoder = spm.SentencePieceProcessor("./ku.spm.model")
def translate(text):
tokens = spm_encoder.encode(text, out_type=str)
result = translator.translate_batch([tokens])
output = spm_decoder.decode(result[0].hypotheses[0], out_type=str)
return output
iface = gr.Interface(
translate,
inputs="text",
outputs="text"
)
iface.launch()