|
--- |
|
license: openrail |
|
metrics: |
|
- bleu |
|
pipeline_tag: text-generation |
|
tags: |
|
- code |
|
--- |
|
|
|
## Story Generation Using GPT-2 in Hugging Face |
|
This repository provides an example of how to use the GPT-2 language model in Hugging Face for story generation tasks. GPT-2 is a powerful natural language processing model that can generate human-like text, and Hugging Face is a popular open-source library for working with NLP models. |
|
|
|
## Requirements |
|
- Python 3.6 or higher |
|
- Hugging Face transformers library |
|
- PyTorch or TensorFlow |
|
|
|
## Installation |
|
- Clone this repository: git clone ```https://github.com/BaoToan1704/Deep-Learning/Final%20Project``` |
|
- Navigate to the repository directory: ```cd Final Project``` |
|
- Install the required libraries: ```pip install -r requirements.txt``` |
|
|
|
## Usage |
|
- Download the GPT-2 pre-trained model: ```python download_model.py``` |
|
- Edit the ```Gpt_2_to_generate_stories.ipynb``` file to include your desired prompt and generate settings. |
|
- Run the ```Gpt_2_to_generate_stories.ipynb file``` to generate text: ```python Gpt_2_to_generate_stories.ipynb``` |
|
|
|
## Customization |
|
You can customize the GPT-2 model and the text generation settings by editing the ```Gpt_2_to_generate_stories.ipynb``` file. For example, you can change the prompt text, the number of tokens to generate, the temperature setting for the model, and more. |
|
|
|
## References |
|
- Hugging Face Transformers library: ```https://github.com/huggingface/transformers``` |
|
- GPT-2 model by me: ```https://huggingface.co/baotoan2002/GPT-2``` |
|
- OpenAI GPT-2 model: ```https://openai.com/models/gpt-2/``` |
|
|
|
## License |
|
This repository is licensed under the [openrail] License. See the LICENSE file for details. |
|
|
|
## Acknowledgments |
|
- Special thanks to the Hugging Face team for their excellent work on the Transformers library. |
|
- Thanks to OpenAI for providing the pre-trained GPT-2 model. |
