From 8cc78d8d7e80a1804bd8229c13738ecf5cb5b8a8 Mon Sep 17 00:00:00 2001 From: ahzero7d1 Date: Fri, 1 Mar 2024 21:47:19 +0900 Subject: [PATCH] politics_serving.py --- projects/bubblow/politics_serving.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/projects/bubblow/politics_serving.py b/projects/bubblow/politics_serving.py index 3031fda..dc74ec1 100644 --- a/projects/bubblow/politics_serving.py +++ b/projects/bubblow/politics_serving.py @@ -3,13 +3,17 @@ import joblib from sklearn.feature_extraction.text import TfidfVectorizer from typing import List +import os # FastAPI 애플리케이션 생성 app = FastAPI() +# 현재 파일의 디렉토리 경로 가져오기 +dir_path = os.path.dirname(os.path.realpath(__file__)) + # 모델 및 TF-IDF 변환기 로드 -model = joblib.load('lgbm_model.joblib') -tfidf_vectorizer = joblib.load('tfidf_vectorizer.joblib') +model = joblib.load(os.path.join(dir_path, 'politics_model/lgbm_model.joblib')) +tfidf_vectorizer = joblib.load(os.path.join(dir_path, 'politics_model/tfidf_vectorizer.joblib')) # 데이터를 받기 위한 Pydantic 모델 정의 class Item(BaseModel):