This repository contains code for fine-tuning a BERT model for sentiment analysis and deploying it as a Streamlit app. The model is trained to classify text into positive or negative sentiment categories.
Before running the code, ensure you have the following packages installed:
!pip install datasets transformers huggingface_hub
!apt-get install git-lfs
The sentiment analysis model used in this code is based on the BERT architecture. It has been fine-tuned on a sentiment analysis dataset. The trained model is available on the Hugging Face Model Hub at Hugging Face Model hub. The streamlit app is available here
The model has been fine-tuned on IMDB Dataset