This repo contains the assignment file I submitted for my Generative AI course hosted by ShiftKey Labs.
The assignment implements a text summarization model using the FLAN-T5 architecture trained on a smaller dataset. The entire process is documented in a Google Colab Notebook, which serves as the assignment file.
Hugging face Model
I uploaded my trained (or fine tuned) model to Hugging Face, which can be accessed here
References:
- Model Used - https://huggingface.co/google/flan-t5-small
- Dataset Used - https://huggingface.co/datasets/abisee/cnn_dailymail
To replicate the project or run your own summarization tasks, you can clone this repository and open the Google Colab Notebook.
git clone https://github.com/varshivenkatesh/flan-t5-text-summarizer.git