Dulanga Heshan

im also facing the same issue, I trained pytorch yolo V5 model, and then tried to integrate with flask API, ``` class Model(object): def __init__(self, model): self.device = torch_utils.select_device() print(self.device)...