Make notebook run on colab

[romeokienzler]

[[Describe the issue
A clear and concise description of what the issue is.

To Reproduce (optional, but appreciated)
Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Screenshots or log output (optional)
If applicable, add screenshots or log output to help explain your problem.

Log Output

Paste the log output here.

Expected behavior (optional)
A clear and concise description of what you expected to happen.

Deployment information (optional)
Describe what you've deployed and how:

• TerraTorch version: [e.g. 1.5.3]
• Installation source: [e.g. git, pip]
  ](https://github.com/NASA-IMPACT/Prithvi-EO-2.0/tree/main/examples/carbon_flux)](https://github.com/NASA-IMPACT/Prithvi-EO-2.0/tree/main/examples/carbon_flux)