ML App implemented with HuggingFace and Streamlit using a finetuned GPT-2 model trained with SpongeBob scripts.
This app uses HuggingFace, an open repository and library to pretrained transformer models. A smaller version of GPT-2 that was pretrained was finetuend on scripts of the show SpongeBob SquarePants, then an app was implemented and hosted with Streamlit.
launch_app.py is the script that launches the app.
The link to the app is here: https://share.streamlit.io/coldestadam/spongebob-text-generation/main/launch_app.py
- Original Pretrained Model: https://huggingface.co/distilgpt2
- Finetuned Model Repo: https://huggingface.co/Coldestadam/Breakout_Mentors_SpongeBob_Model
- Kaggle Dataset: https://www.kaggle.com/mikhailgaerlan/spongebob-squarepants-completed-transcripts