Update app.py

app.py

@@ -1,7 +1,15 @@
-# 가장 상단에 패치를 적용 (모델 로딩 전에 실행)
-from transformers import PreTrainedModel
+# --- 패치 적용: 모델 로딩 전에 실행 ---
+from transformers import AutoModelForImageSegmentation, PreTrainedModel
+
+# 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
@@ -14,7 +22,6 @@ from typing import Tuple, Optional
 
 from PIL import Image
 from gradio_imageslider import ImageSlider
-from transformers import AutoModelForImageSegmentation
 from torchvision import transforms
 
 import requests
@@ -81,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)