CommencezCommencez gratuitement

Charger le modèle d'IA au démarrage du serveur

Vous devez déployer un modèle entraîné d'analyse de sentiment pour aider à modérer les commentaires des utilisateurs. Pour assurer une disponibilité sans interruption, l'API doit être prête à analyser les commentaires dès son démarrage.

Dans cet exercice, vous allez mettre en œuvre les événements de cycle de vie de FastAPI pour charger efficacement votre modèle et bâtir le système de modération des commentaires. La classe de modèle SentimentAnalyzer est déjà définie et importée pour vous.

Cette activité fait partie du cours

Déployer l'IA en production avec FastAPI

Voir le cours

Instructions de l’exercice

  • Importez le décorateur de gestionnaire de contexte du module contextlib pour créer l'événement de cycle de vie.
  • Utilisez le décorateur de gestionnaire de contexte de FastAPI pour définir la fonction d'événement lifespan afin d'assurer le chargement du modèle au démarrage.
  • Appelez la fonction de chargement du modèle au démarrage dans l'événement lifespan.
  • Utilisez yield pour permettre au processus de démarrage du serveur de se poursuivre.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# 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
    ____
    ____
Modifier et exécuter le code