get_predictions.py has an error if we use CUDA

#2
by Yehor - opened

Sign up or log in to comment