Agentic Transformer System for Hugging Face
This repository contains the code, configurations, and scripts to create a modular AI system with multiple specialized agents for text classification, sentiment analysis, summarization, code generation, and more.
Project Overview
This system uses a mixture of specialized agents to handle diverse tasks. Each agent is fine-tuned for its specific role and is managed by an orchestration layer that ensures smooth communication and data flow between agents.
Models and Datasets
Pre-trained Models
- Text Classification Agent: BERT Base Uncased (Hugging Face Model)
- Sentiment Analysis Agent: RoBERTa (Hugging Face Model)
- Summarization Agent: BART (Hugging Face Model)
- Code Generation Agent: CodeBERT (Hugging Face Model)
- Web Search Agent: DistilBERT (Hugging Face Model)
Datasets
- Training and Validation Dataset: IMDB Movie Reviews (Hugging Face Dataset)
- Evaluation Dataset: SST-2 (Hugging Face Dataset)
Installation
To install the required dependencies, run:
pip install -r requirements.txt