This project trains BERT and Seq2Seq models on a Hinglish dataset.
Create a virtual environment:
python3 -m venv .venv
source .venv/bin/activate
Install dependencies:
pip install -r requirements.txt
The dataset is located in the data/ directory:
train.tsvvalidation.tsvtest.tsvOpen project.ipynb in VS Code or Jupyter Lab to run the training pipeline.
The notebook covers:
bert-base-multilingual-cased to classify errors (e.g., Grammar, Spelling, Slang).google/mt5-small to correct Hinglish sentences.To run the Grammarly-like UI:
streamlit run app.py
The trained models will be saved in:
bert-domain-classifier/mt5-hinglish-correction/