UNetR Architecture Implementation with Multi-Class prediction capability