If you want to display category in media, use this plugin. It adds media category option in WordPress admin dashboard Media Menu. WordPress Media Category is very useful plugin for creating media category in WordPress.