import gradio as gr
import torch

from germantoenglish import Seq2SeqTransformer, translate, greedy_decode 


if __name__ == "__main__":
    iface = gr.Interface(
        fn=translate, 
        inputs=[
            gr.components.Textbox(label="Text")

    ],
    outputs=["text"],  
    cache_examples=False, 
    title="GermanToEnglish",  
    )

    iface.launch(share=True)