Hi,
You can assign the categories as done in default Posts.
Add two categories, ex 1) free 2) premium. Assign category to the file accordingly. (Attached a screenshot, which may help you.) Then use category short code. use Advanced TinyMCE button plugin to help you to check short-codes.
Thank you.