ai based restoration