Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-os.environ["HF_FORCE_SAFE_SERIALIZATION"] = "1"   # ✅ safetensors만 사용하도록 강제
+os.environ["HF_FORCE_SAFE_SERIALIZATION"] = "1"   # 이미 안전모드 # ✅ safetensors만 사용하도록 강제
 
 import gradio as gr
 from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoTokenizer
@@ -11,7 +11,9 @@ import tempfile
 
 # ───── 이미지 캡셔닝 모델 로딩 ─────
 caption_model = VisionEncoderDecoderModel.from_pretrained(
-    "nlpconnect/vit-gpt2-image-captioning"
+    "nlpconnect/vit-gpt2-image-captioning",
+    use_safetensors=True,                 # ✅ 올바른 옵션명
+    low_cpu_mem_usage=True               # (옵션) 메모리 절약
 )
 feature_extractor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")