Bắt đầu ngayBắt đầu miễn phí

Nạp AI model khi máy chủ khởi động

Bạn cần triển khai một model phân tích cảm xúc đã được huấn luyện để hỗ trợ kiểm duyệt bình luận của người dùng. Để đảm bảo không có thời gian chết, API phải sẵn sàng phân tích bình luận ngay khi khởi động.

Trong bài tập này, bạn sẽ triển khai lifespan events của FastAPI để nạp model một cách hiệu quả, phục vụ hệ thống kiểm duyệt bình luận. Lớp model SentimentAnalyzer đã được định nghĩa và import sẵn cho bạn.

Bài tập này là một phần của khóa học

Triển khai AI vào môi trường sản xuất với FastAPI

Xem khóa học

Hướng dẫn bài tập

  • Import decorator của context manager từ module contextlib để tạo lifespan event.
  • Sử dụng decorator context manager của FastAPI để định nghĩa hàm sự kiện lifespan nhằm đảm bảo model được nạp khi khởi động.
  • Gọi hàm để nạp model khi khởi động trong sự kiện lifespan.
  • Gọi yield để quá trình tải máy chủ tiếp tục.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

# 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
    ____
    ____
Chỉnh sửa và Chạy Mã