Update app.py

app.py

@@ -34,9 +34,10 @@ POSTGRES_DB = os.getenv('POSTGRES_DB')
 GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
 
 COLLECTION_NAME = "homepage_pdfplumner_1st"
+SENTENCE_TRANSFORMERS_HOME = os.getenv('SENTENCE_TRANSFORMERS_HOME')
 
 # 2. 필수 환경 변수가 모두 존재하는지 확인합니다.
-if not all([POSTGRES_USER, POSTGRES_PASSWORD, POSTGRES_HOST, POSTGRES_PORT, POSTGRES_DB, COLLECTION_NAME, GOOGLE_API_KEY]):
+if not all([POSTGRES_USER, POSTGRES_PASSWORD, POSTGRES_HOST, POSTGRES_PORT, POSTGRES_DB, GOOGLE_API_KEY, SENTENCE_TRANSFORMERS_HOME]):
     raise ValueError("필수 환경 변수들이 설정되지 않았습니다. Hugging Face Secrets를 확인하세요.")
 
 # 환경 변수를 조합하여 CONNECTION_STRING을 생성
@@ -61,7 +62,8 @@ app.add_middleware(
 # RAG 구성 요소를 프로그램 시작 시 한 번만 초기화
 embeddings = HuggingFaceEmbeddings(
     model_name='nlpai-lab/KURE-v1',
-    model_kwargs={'device': 'cpu'}
+    model_kwargs={'device': 'cpu'},
+    cache_folder=SENTENCE_TRANSFORMERS_HOME
 )
 
 try: