1a6d4020c53e2b39e224616481528fad901cbd4e,deeppavlov/core/commands/infer.py,,interact_model,#Any#,95

Before Change

            if args[-1] == "exit" or args[-1] == "stop" or args[-1] == "quit" or args[-1] == "q":
                return

        if len(args) == 1:
            pred = model(args)
        else:
            pred = model([args])

        print(">>", *pred)


def predict_on_stream(config_path: str, batch_size: int = 1, file_path: Optional[str] = None) -> None:

After Change

        pred = model(*args)
        if len(model.out_params) > 1:
            pred = zip(*pred)

        print(">>", *pred)

Instances