streamlit_apps/app_utils/device.py

import torch

cpu_device = torch.device("cpu")
# device = torch.device("cuda:3" if torch.cuda.is_available() else "cpu")
device = cpu_device