They are both OK. "what" is more informal and more common in this context. "which" is grammatically more correct but still OK to use informally too.