AI could predict who will have a heart attack