Replaced typing module generics like List, Dict, Tuple with built-in alternatives (list, dict, tuple).
Updated code to use new union syntax (X | Y) instead of Union[X, Y].
Added arguments for creating only prediction masks, as well as refusing to save the output;
Added execution time display;
The pretrained_weight parameter has been moved to common;
Parameters related to ensembles have been removed.