Update app.py

app.py

@@ -1,12 +1,15 @@
 # --- 패치 적용: 모델 로딩 전에 실행 ---
 from transformers import PretrainedConfig
 PretrainedConfig.get_text_config = lambda self, decoder=True: type("DummyTextConfig", (), {"tie_word_embeddings": False})()
-# --- 패치 종료 ---
 
-from transformers import AutoModelForImageSegmentation, PreTrainedModel
-# tie_weights를 빈 함수로 오버라이드 (모델 로딩 시 tie_weights 호출 무시)
-PreTrainedModel.tie_weights = lambda self: None
+# 이미지 분할 모델에 해당하는 모든 클래스의 tie_weights를 빈 함수로 오버라이드
+from transformers.models.auto.modeling_auto import MODEL_FOR_IMAGE_SEGMENTATION_MAPPING
+for model_class in MODEL_FOR_IMAGE_SEGMENTATION_MAPPING.values():
+    model_class.tie_weights = lambda self: None
+# --- 패치 종료 ---
 
+from transformers import AutoModelForImageSegmentation
+from transformers import PreTrainedModel  # (참고용)
 import os
 import cv2
 import numpy as np
@@ -85,7 +88,7 @@ usage_to_weights_file = {
 
 # 초기 모델 로딩 (기본: General)
 birefnet = AutoModelForImageSegmentation.from_pretrained(
-    '/'.join(('zhengpeng7', usage_to_weights_file['General'])), 
+    '/'.join(('zhengpeng7', usage_to_weights_file['General'])),
     trust_remote_code=True
 )
 birefnet.to(device)