fix: avoiding error while --device is "cpu"

[queuecameras]

Title of PR:

Make code device-agnostic by replacing cuda() with to(device)

---

Description:

This PR refactors the code to make it device-agnostic. Specifically, it replaces the hardcoded .cuda() call with the more flexible .to(device) method. This change ensures that the code can run seamlessly on either a CPU or a GPU, depending on the environment.

Changes made:

• Introduced a device variable that checks if a GPU is available and sets the device accordingly.
• Replaced the .cuda() call with .to(device) to move the tensor to the appropriate device.

Benefits:

• Flexibility: The code can now run on either CPU or GPU without modification.
• Compatibility: Ensures compatibility across different environments, making it easier to run the code on various machines without requiring manual changes.