The model trained in the /training folder has been fine-tuned from a distilled version of BERT in order to classify articles by theme (business, politics, tech, entertainment, sport) and has been uploaded to: https://huggingface.co/srinivajan/Classification
To try it, either run demo.py and visit http://127.0.0.1:7860, or just run the single cell in the notebook demo.ipynb
