| import gradio as gr | |
| from fairseq.models.transformer import TransformerModel | |
| model_directory='./' | |
| model = TransformerModel.from_pretrained( | |
| model_directory, | |
| checkpoint_file='model.pt', | |
| data_name_or_path='./bin', | |
| bpe='sentencepiece', | |
| sentencepiece_model='./spm.model' | |
| ) | |
| def translate(text): | |
| return model.translate(text) | |
| iface = gr.Interface(fn=translate, inputs="text", outputs="text") | |
| iface.launch() | |