TASTY: A Transformer based Approach to Space and Time complexity