ChatGPT-2-Explorer is a conversational AI project based on GPT-2 architecture, offering tools for training, testing, and exploring a chatbot to gain insights into GPT-2’s capabilities. Designed for adaptability with custom datasets, this repository demonstrates fine-tuning methods on conversational data, allowing the model to generate interactive and contextually relevant responses.
- 🔄 Custom Dataset Training: Easily adapt GPT-2 for specific conversational datasets.
- 💬 Interactive Chatbot Interface: Engage in real-time conversations with your model.
- ⚙️ GPT-2 Customization: Experiment with model parameters for tailored performance.
trainig.py
: Code for fine-tuning GPT-2.chatbot.py
: Script for running the chatbot.torch examples.ipynb
: PyTorch usage examples.vocab.txt
: Tokenization vocabulary.wizard_of_oz.txt
: Sample conversational dataset.
- Clone the Repository:
git clone https://github.com/jaidh01/ChatGPT-2-Explorer.git
- Train the Model:
python trainig.py
- Run the Chatbot:
python chatbot.py
- Python 3.8+
- PyTorch
- Transformers library
After running chatbot.py
, converse with the model directly via the terminal.
- 💻 Web-based UI for a more user-friendly experience
- 📈 Support for larger datasets
Contributions are welcome! Feel free to fork, improve, and submit a pull request. 😊