from transformers import BartTokenizer
from builders import SourcesBuilder
from fetchers import fetch_seq2seq


def main():
    model = fetch_seq2seq("overfit")
    tokenizer = BartTokenizer.from_pretrained("facebook/bart-base")
    lit2idi = [
        ("my man", ""),
        ("hello", "")
    ]  # just some dummy stuff
    srcs = SourcesBuilder(tokenizer)(lit2idi)
    out = model.predict(srcs=srcs)
    print(out)


if __name__ == '__main__':
    main()