Malek NASKI
Add application file
6ac8934
import gradio as gr
import os
def pred(text):
model = "averaged-10-epoch.pt"
hf_repo_id = "malloc/OpenNMT-py-English-German-Transformer"
src = "src-test.txt"
with open(src, 'w') as f:
f.write(text)
output = "pred.txt"
os.system(f"onmt_translate -model {model} -hf_repo_id {hf_repo_id} -src {src} -output {output}")
with open(output) as f:
translation = f.read()
return translation
iface = gr.Interface(
fn=pred,
inputs= gr.inputs.Textbox(lines=2, placeholder="Enter English text to translate..."),
outputs=["text"],
theme="huggingface")
iface.launch()