Missing CUDA availability check in ESM-1v example

[ebeurer]

In examples/variant-prediction/predict.py, there are some instances where CUDA is used without checking availability (and the nogpu argument), namely lines 142, 175, 194 and 212.

In line 155 those checks are performed before using CUDA, which leads me to believe the code is intended to also work when using only a CPU.