Chargement du modèle d'IA au démarrage du serveur
Il est nécessaire de déployer un modèle d'analyse des sentiments bien formé qui facilite la modération des commentaires des utilisateurs. Afin de garantir une disponibilité totale, l'API doit être prête à analyser les commentaires des utilisateurs dès son démarrage.
Dans cet exercice, vous allez implémenter les événements de durée de vie de FastAPI afin de charger efficacement votre modèle et de construire les systèmes de modération des commentaires. La classe modèle SentimentAnalyzer
est déjà définie et importée pour vous.
Cet exercice fait partie du cours
Déployer l'IA en production avec FastAPI
Instructions
- Importez le décorateur context manager à partir du module
contextlib
afin de créer l'événement lifespan. - Veuillez utiliser le décorateur context manager de FastAPI pour définir la fonction d'événement
lifespan
afin de garantir le chargement du modèle au démarrage. - Appelez la fonction pour charger le modèle au démarrage dans l'événement
lifespan
. - Céder pour permettre au processus de chargement du serveur de se poursuivre.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Import the context manager decorator from contextlib module
from contextlib import ____
sentiment_model = None
def load_model():
global sentiment_model
sentiment_model = SentmentAnalyzer("sentiment_model.joblib")
# Use FastAPI's context manager to define lifespan event
@____
def lifespan(app: FastAPI):
# Call the function to load the model
____
____