def load_model(self):
    model = torch.hub.load(self.hub_repo, 'MiDaS', pretrained=True)
    model.to(self.device)
    model.eval()
    return model
def load_model(self):
    model = torch.hub.load(self.hub_repo, 'MiDaS', pretrained=True)
    model.to(self.device)
    model.eval()
    return model