def test_serve(): model = TextClassifier(2, TEST_BACKBONE) # TODO: Currently only servable once a preprocess and postprocess have been attached model._preprocess = TextClassificationPreprocess(backbone=TEST_BACKBONE) model._postprocess = TextClassificationPostprocess() model.eval() model.serve()
def test_serve(): model = TextClassifier(2, backbone=TEST_BACKBONE) model.eval() model.serve()