Update src/generate_caption.py

src/generate_caption.py

@@ -4,21 +4,18 @@ import torch
 import os
 import streamlit as st
 
-# ✅ HF 캐시 경로 (쓰기 가능한 임시 경로로 변경)
-HF_CACHE_DIR = "/tmp/hf_cache"
-os.environ["HF_HOME"] = HF_CACHE_DIR
-os.environ["TRANSFORMERS_CACHE"] = HF_CACHE_DIR
+
+# 안전한 캐시 디렉토리 지정
+HF_CACHE_DIR = os.path.join(tempfile.gettempdir(), "hf_cache")
 os.makedirs(HF_CACHE_DIR, exist_ok=True)
 
-@st.cache_resource(show_spinner="BLIP 모델 로딩 중...")  # ✅ 사용자 피드백
-def load_blip_model():
-    processor = BlipProcessor.from_pretrained(
-        "Salesforce/blip-image-captioning-base", cache_dir=HF_CACHE_DIR
-    )
-    model = BlipForConditionalGeneration.from_pretrained(
-        "Salesforce/blip-image-captioning-base", cache_dir=HF_CACHE_DIR
-    )
-    return processor, model
+# 환경 변수 설정 (ONLY HF_HOME)
+os.environ["HF_HOME"] = HF_CACHE_DIR
+
+# transformers.load에서 cache_dir 지정
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base", cache_dir=HF_CACHE_DIR)
+model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base", cache_dir=HF_CACHE_DIR)
+
 
 def generate_caption(image_path):
     processor, model = load_blip_model()