def METHOD_NAME(cls):
    METHOD_NAME = super().METHOD_NAME()
    METHOD_NAME.update({
        'model': PathField(description="Path to model.", file_or_directory=True),
        'weights': PathField(description="Path to weights.", optional=True, file_or_directory=True),
        'device': StringField(regex=DEVICE_REGEX, description="Device name."),
        'batch': NumberField(
            value_type=int, min_value=1, optional=True, default=1, description="Batch size."
        ),
        'allow_reshape_input': BoolField(optional=True, default=False, description="Allows reshape input.")
    })
    return METHOD_NAME