Skip to content

Shreyansh-1998/TextCondenserAI

Repository files navigation

Text Condensation using Transformers Hugging Face API

This project focuses on Text Condensation giving a summary of the entire conversation in this case dialogues, future integration may include Machine Learning Operations (MLOps) practices and Data Version Control (DVC) for efficient project management and versioning. The core of the solution is built around the Transformers. Pytorch library with Transformers Hugging Face API with Google model is used for efficient model training. It is trained on both C4 and HugeNews and is trained for 1.5M to observe slower convergence on pretraining complexity. Dataset source.

How to run?

Clone the repository

```
git clone https://github.com/Shreyansh-1998/TextCondenserAI.git
```

Create a conda environment

```
conda create -n your_env_name -python=3.8 -y
```
```
conda activate your_env_name
```

Install the requirements

```
pip install -r requirements.txt
```

Run the main file

```
python main.py
```

About

This project focuses on NLP giving a summary of the entire conversation using google/pegasus model

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages