Update app.py

app.py

@@ -8,7 +8,10 @@ import os
 import tempfile
 
 # ───── 이미지 캡셔닝 모델 로딩 ─────
-caption_model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
+caption_model = VisionEncoderDecoderModel.from_pretrained(
+    "nlpconnect/vit-gpt2-image-captioning", 
+    safetensors=True  # ✅ torch.load() 대신 안전한 방식 사용
+)
 feature_extractor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")