Skip to content

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.

Notifications You must be signed in to change notification settings

jaidh01/ChatGPT-2-Explorer

Repository files navigation

🌐 ChatGPT-2-Explorer

📚 Overview

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.

🚀 Features

  • 🔄 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.

📂 Files

  • 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.

🛠️ Getting Started

  1. Clone the Repository:
    git clone https://github.com/jaidh01/ChatGPT-2-Explorer.git
  2. Train the Model:
    python trainig.py
  3. Run the Chatbot:
    python chatbot.py

✅ Requirements

  • Python 3.8+
  • PyTorch
  • Transformers library

💬 Example Usage

After running chatbot.py, converse with the model directly via the terminal.

🌱 Future Plans

  • 💻 Web-based UI for a more user-friendly experience
  • 📈 Support for larger datasets

🤝 Contributions

Contributions are welcome! Feel free to fork, improve, and submit a pull request. 😊

About

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.

Resources

Stars

Watchers

Forks