Transfer Learning Model hosted on Heroku using React & Flask

Transfer Learning Model using RoBERTa/LSTM built on PyTorch, trained on the IMDb dataset hosted on Heroku using React and Flask.

Comment 1

posted an update —

New Features Added

  • Three different sections have been created for the different parts of the tutorials, these are divided into chapters for a better understanding of the project.
  • A starter project has been added using a simple PyTorch model using the LSTM layer for beginners.
  • Colab notebook for the RoBERTa model has been added to provide a step by step guide for the tutorial
  • The UI for the web application has been revamped to make it easier to access.

Log in or sign up for Devpost to join the conversation.