Conversation AI Bot Ainz-Ooal-Gown (Fine-tuned DialoGPT Model)

Overview

• This project is an AI bot.
• The model used is "DialoGPT" from Microsoft
• The model is trained with custom dataset to act like the character "Ainz Ooal Gown" from the anime "Overlord".
• Model has been trained using Transformers and HuggingFace.

Features

• Train any AI Model (DialoGPT in this case).
• Train on any Character's dataset (Ainz Ooal Gown in this case).
• Use Huggingface's API to interact with Model.

Requirements

• Python
• VS Code (Or Any Other IDE)
• Huggingface

Usage

• Clone the repository and open with VS Code (or any other suitable IDE).
• Train the model locally or on google colab.
• This Repository has 2 Python files-
  • no_context.py - The AI Bot Doesn't Remember Conversations.
  • with_context.py - The AI Bot Remembers Conversstions.

Screenshots

Conversation with AI