https://github.com/cdpierse/script_buddy_v2

Script Buddy v2 is a film script text generation tool built using film scripts from the world's most popular film scripts and GPT2.
https://github.com/cdpierse/script_buddy_v2

artificial-intelligence gpt-2 language-generation machine-learning nlp pytorch transformers

Last synced: 7 days ago
JSON representation

Script Buddy v2 is a film script text generation tool built using film scripts from the world's most popular film scripts and GPT2.

• Host: GitHub
• URL: https://github.com/cdpierse/script_buddy_v2
• Owner: cdpierse
• License: mit
• Created: 2019-12-18T23:48:54.000Z (almost 5 years ago)
• Default Branch: master
• Last Pushed: 2023-07-23T00:46:03.000Z (over 1 year ago)
• Last Synced: 2024-11-01T04:51:40.385Z (14 days ago)
• Topics: artificial-intelligence, gpt-2, language-generation, machine-learning, nlp, pytorch, transformers
• Language: Jupyter Notebook
• Homepage:
• Size: 83.8 MB
• Stars: 47
• Watchers: 9
• Forks: 22
• Open Issues: 8
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# script_buddy_v2

 Script Buddy v2 is a film script text generation tool built using film scripts from the world's most popular film scripts. 

 Read more about the project on [Medium](https://towardsdatascience.com/film-script-generation-with-gpt-2-58601b00d371?source=friends_link&sk=934ac3ac7079d34bae215ce9a558986a) 

 I'm also a bot on twitter, I tweet a new script snippet every hour or so, follow me at [@script_buddy](https://twitter.com/script_buddy)

![Script Buddy APP](script_buddy/images/app.gif)

 

## Project Structure 

- *imsdb_scraper* contains the scrapy spider built to crawl IMSDB.com and download the film script data that is used to train our text generation model. 

    - To run the scraper change directory to imsdb_scraper with `cd imsdb_scraper`. As I haven't built any pipelines currently to run the scraper and save it's output simply run `scrapy crawl scriptSpider  -o scripts.json -t json` and for now manually move this output file to `script_buddy/data/` and you'll be good to go. 

- *script_buddy* contains the model fine tuning notebook, utility script parsing functions, tweepy bot code, and streamlit application code. 

    - If you'd like to run the streamlit app locally 

        - `git clone https://github.com/cdpierse/script_buddy_v2.git` 

        - `pip install -r requirements.txt`

        - `cd script_buddy`

        - `streamlit run app.py`

    - This should open an instance of the app for you to experiment with. Running it for the first time may take a 

    while as the model needs to be downloaded from huggingface's model hub.