Your predict_covid method seems to return a string, so you would have to look into this method and check, what it is doing.
The standard PyTorch models return e.g. class logits and you could get the predicted class index via: preds = torch.argmax(outputs, dim=1) for a multi-class classification.