Prompt Classification
Aim:
- Develop a lightweight prompt classification system using a relatively small large language model. (upper limit of 8 billion parameters).
- Design capability to automatically route classified prompts to their corresponding models (text/image generation).
- Integrate customizable guardrails to filter unethical, harmful or restricted prompts.
For instructions on how to set up Project Environment, please refer to environment set up instructions pdf.
Disclaimer: create a .env file for api_key, or put it within the notebook for one-time, short demonstration purposes.
System Design
