refactor output tracing in upernet

[omkar-334]

This PR refactors the upernet model as per #43979

cc @molbap

[omkar-334]

I don't know why this test is failing -

data: {
      message: 'No commit found for SHA: ',
      documentation_url: 'https://docs.github.com/rest/commits/commits#get-a-commit',
      status: '422'
    }

Maybe you could re-run it to see if it passes.

[github-actions]

[For maintainers] Suggested jobs to run (before merge)

run-slow: upernet