Tips for Choosing Model: Automatic Labelling of Articles

I am looking for an advice in choosing the right model to perform

  1. Automatic generation of labels in an Article
    Any models which does “title” generation for articles can also be helpful

** Preferably I need some light-weight model to run it in my laptop **

Any suggestions are welcome!

You could try to use Seq2Seq models for this (BART, T5 etc) where input seq would be the article and output seq the title. But both these models are large, T5 has a small variant t5-small which is around 230MB.

1 Like