[8.x] [ML] File upload: enabling check for model allocations (#197395)

[8.x] [ML] File upload: enabling check for model allocations (#197395) #9647