Update app.py

app.py

@@ -1,15 +1,12 @@
 # --- 패치 적용: 모델 로딩 전에 실행 ---
-from transformers import AutoModelForImageSegmentation, PreTrainedModel
+from transformers import PretrainedConfig
+PretrainedConfig.get_text_config = lambda self, decoder=True: type("DummyTextConfig", (), {"tie_word_embeddings": False})()
+# --- 패치 종료 ---
 
-# tie_weights를 빈 함수로 오버라이드 (이미 존재하는 경우)
+from transformers import AutoModelForImageSegmentation, PreTrainedModel
+# tie_weights를 빈 함수로 오버라이드 (모델 로딩 시 tie_weights 호출 무시)
 PreTrainedModel.tie_weights = lambda self: None
 
-# 모델의 config 클래스에 dummy get_text_config 메서드를 추가
-def dummy_get_text_config(self, decoder=True):
-    return type("DummyTextConfig", (), {"tie_word_embeddings": False})()
-AutoModelForImageSegmentation.config_class.get_text_config = dummy_get_text_config
-# --- 패치 종료 ---
-
 import os
 import cv2
 import numpy as np
@@ -88,7 +85,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)