File size: 471 Bytes
056f7e1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import joblib
from fastapi import HTTPException

def load_model_and_vectorizer(model_path, vectorizer_path):
    try:
        model = joblib.load(model_path)
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Failed to load model: {e}")

    try:
        vectorizer = joblib.load(vectorizer_path)
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Failed to load vectorizer: {e}")

    return model, vectorizer