Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- 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-
|
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
|