Correctly pass tensor to device

According to the docs, input_tensor.to(device) doesn't move the tensor to the device but returns a tensor that's on the device.

Fixes GPU predictions

changed milestone to %ML Prod - March 2023 n°2

added P1 label

assigned to @schneider-y

added 2 commits

• 6f97aa8f - 1 commit from branch main
• c42e3502 - correctly pass tensor to device

Compare with previous version

enabled an automatic merge when the pipeline for c42e3502 succeeds

merged