This project implements a text summarization system using state-of-the-art natural language processing techniques.
- Automatic text summarization
- Fine-tuning of pre-trained language models
- Evaluation metrics for summarization quality
- Support for both extractive and abstractive summarization
- Python 3.x
- PyTorch
- Transformers library (Hugging Face)
- NLTK (Natural Language Toolkit)
- ROUGE (Recall-Oriented Understudy for Gisting Evaluation) for evaluation
train.py: Main script for training the summarization model
-
Clone the repository:
git clone https://github.com/your-username/Text_summarizer.git cd Text_summarizer -
Create a virtual environment (optional but recommended):
python -m venv venv source venv/bin/activate # On Windows, use `venv\Scripts\activate` -
Install the required dependencies:
pip install -r requirements.txt
-
Prepare your dataset in the required format.
-
Run the training script:
python train.py -
For inference, use:
python inference.py --input "Your text to summarize"
Contributions are welcome! Please feel free to submit a Pull Request.
This project is licensed under the GNU Affero General Public License v3.0 (AGPL-3.0) - see the LICENSE file for details.