What's needed for people in that situation, who are charged with committing a crime while they are waiting for a psychiatric assessment is a forensic psychiatry ward. Depending on the nature of the crime, the regular psych ward may not be able to handle the person.