Update app.py

app.py

@@ -62,10 +62,13 @@ except ModuleNotFoundError:
 # ─────────────────────────────────────────────────────────────
 # 4. 이미지 캡셔닝 모델
 # ─────────────────────────────────────────────────────────────
+# 4. 이미지 캡셔닝 모델 ------------------------------------
 caption_model = VisionEncoderDecoderModel.from_pretrained(
     "nlpconnect/vit-gpt2-image-captioning",
-    use_safetensors=True      # OK
-).eval()                      # 평가 모드로
+    use_safetensors=True,        # 그대로
+    low_cpu_mem_usage=False,     # ← meta 로딩 비활성화
+    device_map=None              # ← Accelerate 자동 분할 끄기
+).eval()                         # 평가 모드
 
 feature_extractor = ViTImageProcessor.from_pretrained(
     "nlpconnect/vit-gpt2-image-captioning"