Rick and Morty Story Generator
This project uses a pre-trained GPT2 model, which was fine-tuned on Rick and Morty transcripts, to generate new stories in the form of a dialog. The project uses Hugging Face's Transformers library to do inference and Streamlit for the application.
Try out the demo to generate fun stories or read the blog post on how to create your own story generator.
Fine-tuning a custom model
You can fine-tune your own model using Google Colab
Setup
This repository has only been tested with Python 3.7.
Install the dependencies in a virtual environment:
python3.7 -m venv venv
source venv/bin/activate
pip install --upgrade pip setuptools
pip install -r requirements.txt
On the first run, the app will download the pre-trained model from Hugging Face's Model Hub or you can supply your own custom model in the load_model() function. To start the application, simply run:
streamlit run app.py
3 Jan 12, 2022
2 Nov 17, 2021
276 Dec 31, 2022
148 Dec 26, 2022
1 Dec 21, 2021
4 Nov 25, 2022
840 Jan 4, 2023
3 Apr 5, 2022
1 Jan 14, 2022
2.3k Dec 29, 2022
785 Dec 29, 2022
1 Jun 11, 2022
1 Jan 6, 2022
11 Jul 21, 2022
9 Jan 1, 2023
4 May 1, 2022
172 Dec 29, 2022
1 Jan 1, 2022