Article Two of the United States Constitution
If the President declares himself unable to discharge his duties, he can take the Presidency back at any time.  For example, in 2002, George W. Bush became the first President to formally use the Disability Clause.  He gave power to his Vice President for about two hours while he had a medical test which required anesthesia.  Once he felt ready to be President again, Bush took back the Presidency.