import gradio as gr
import torch

from germantoenglish import Seq2SeqTransformer, translate, greedy_decode 

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = model.load_state_dict(torch.load('./transformer_model.pth', map_location=device))
model.eval()

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)