Interactive Story Generation

The work in this project has been done by me as my M.Tech Project from IIT Bhubaneswar.

Author: Sai Krishna Jupally

The files in the repo contain:

1. Dataset files
2. Dataset formatter code files for custom use
3. Loss Plot code files with output cells containing GPT2 finetuned model plots

For unsloth models finetuning:

Sample code can be found here

Finetuned models:

Finetuned models (and dataset) can be found at Hugging Face

Sample stories:

Sample stories from finetuned models can be found here

Interactive Story Generator:

This link contains code for:

1. Loading models for inference
2. Interactive Story Generation
3. 150 stories generator by self-feeding lines to the model

Compare models - using Entity based Local Coherence

Code for coherence computation and model comparison available here

Web application

Web app client code available here