Phoenix21 commited on
Commit
0b20500
·
verified ·
1 Parent(s): 756269e

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +0 -3
pipeline.py CHANGED
@@ -25,9 +25,6 @@ from prompts import classification_prompt, refusal_prompt, tailor_prompt
25
  mistral_api_key = os.environ.get("MISTRAL_API_KEY")
26
  client = Mistral(api_key=mistral_api_key)
27
 
28
- # Initialize Pydantic AI Agent (for text validation)
29
- pydantic_agent = Agent('mistral:mistral-large-latest', result_type=str)
30
-
31
  # Load spaCy model for NER and download it if not already installed
32
  def install_spacy_model():
33
  try:
 
25
  mistral_api_key = os.environ.get("MISTRAL_API_KEY")
26
  client = Mistral(api_key=mistral_api_key)
27
 
 
 
 
28
  # Load spaCy model for NER and download it if not already installed
29
  def install_spacy_model():
30
  try: