Phoenix21 commited on
Commit
0dfcfda
·
verified ·
1 Parent(s): fc48f50

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -32,7 +32,7 @@ mistral_client = Mistral(api_key=os.environ["MISTRAL_API_KEY"])
32
  # Initialize Mistral agent using Pydantic AI
33
 
34
  mistral_api_key = os.environ.get("MISTRAL_API_KEY") # Ensure your Mistral API key is set
35
- mistral_model = MistralModel("mistral-large-latest", api_key=mistral_api_key) # Use a Mistral model
36
  mistral_agent = Agent(mistral_model)
37
 
38
  # Load spaCy model for NER and download the spaCy model if not already installed
 
32
  # Initialize Mistral agent using Pydantic AI
33
 
34
  mistral_api_key = os.environ.get("MISTRAL_API_KEY") # Ensure your Mistral API key is set
35
+ mistral_model = MistralModel("mistral-moderation-latest", api_key=mistral_api_key) # Use a Mistral model
36
  mistral_agent = Agent(mistral_model)
37
 
38
  # Load spaCy model for NER and download the spaCy model if not already installed