This project is a Flask-based web application that integrates a fine-tuned language model to provide conversational AI capabilities. The application uses HTML, CSS, and JavaScript for the front-end, and Python for the back-end, leveraging the transformers
library for model handling.
- Interactive chat interface with a chatbot.
- Fine-tuned language model for generating responses.
- Dockerized deployment for easy setup and scaling.
- Dark-themed UI with responsive design.
- Python 3.7 or higher
- Docker (optional, for containerized deployment)
- NVIDIA GPU with CUDA (optional, for faster inference)
git clone https://github.com/zacharias1219/finetuned-custom-llm-model.git
cd finetuned-custom-llm-model
python -m venv venv
source venv/bin/activate # On Windows use `venv\Scripts\activate`
pip install -r requirements.txt
Place your fine-tuned model files in a directory named finetuned_model
within the project directory. Ensure the directory contains the following files:
config.json
generation_config.json
model-00001-of-00006.safetensors
model-00002-of-00006.safetensors
model-00003-of-00006.safetensors
model-00004-of-00006.safetensors
model-00005-of-00006.safetensors
model-00006-of-00006.safetensors
model.safetensors.index.json
special_tokens_map.json
tokenizer_config.json
tokenizer.json
tokenizer.model
python app.py
Open your web browser and navigate to http://127.0.0.1:5000
to interact with the chatbot.
docker build -t chatbot-app .
docker run -p 5000:5000 chatbot-app
Open your web browser and navigate to http://127.0.0.1:5000
to interact with the chatbot.
- Start the application by running
python app.py
or using Docker as described above. - Open your web browser and go to
http://127.0.0.1:5000
. - Type your message in the chat interface and press the send button to interact with the chatbot.
├── app.py # Main application file
├── Dockerfile # Dockerfile for containerized deployment
├── requirements.txt # Python dependencies
├── templates
│ └── chat.html # HTML template for the chat interface
├── static
│ ├── styles.css # CSS file for styling
│ └── images # Directory for image assets
└── finetuned_model # Directory containing the fine-tuned model files
This project is licensed under the MIT License. See the LICENSE file for more details.
There are some file that are missing as it is too big to upload on github so here are the photos