AI model shows promise for diagnosing prostate cancer