predict.py with using gpu

[batooooo]

Hi,
I want to get results using gpu, what should i do?

[saahiluppal]

Hi,
move model to cuda model.to(device)
move input image to cuda input.to(device)