def test_serve(): model = TranslationTask(TEST_BACKBONE) # TODO: Currently only servable once a preprocess and postprocess have been attached model._preprocess = TranslationPreprocess(backbone=TEST_BACKBONE) model._postprocess = Seq2SeqPostprocess() model.eval() model.serve()
def test_serve(): model = TranslationTask(TEST_BACKBONE) model.eval() model.serve()