Cannot run on Colab

I ran this Colab file "https://colab.research.google.com/drive/1vyqbBzBDpgNhjGGuGsiVJb336-WDunOv?usp=sharing", and I performed the environment setup and setup process smoothly. However, when it came to the inference step, it encountered this error.

Traceback (most recent call last):
File "/usr/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/usr/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
File "/content/alphageometry/alphageometry.py", line 26, in
import ddar
File "/content/alphageometry/ddar.py", line 20, in
import dd
File "/content/alphageometry/dd.py", line 24, in
import graph as gh
File "/content/alphageometry/graph.py", line 31, in
import numericals as nm
File "/content/alphageometry/numericals.py", line 30, in
matplotlib.use('TkAgg')
File "/usr/local/lib/python3.10/dist-packages/matplotlib/cbook/deprecation.py", line 296, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/matplotlib/cbook/deprecation.py", line 358, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/matplotlib/__init__.py", line 1281, in use
plt.switch_backend(name)
File "/usr/local/lib/python3.10/dist-packages/matplotlib/pyplot.py", line 234, in switch_backend
raise ImportError(
ImportError: Cannot load backend 'TkAgg' which requires the 'tk' interactive framework, as 'headless' is currently running