Update app.py

app.py

@@ -10,7 +10,7 @@ from typing import Tuple, Optional
 
 from PIL import Image
 from gradio_imageslider import ImageSlider
-from transformers import AutoModelForImageSegmentation, PretrainedConfig
+from transformers import AutoModelForImageSegmentation, PreTrainedModel
 from torchvision import transforms
 
 import requests
@@ -18,8 +18,8 @@ from io import BytesIO
 import zipfile
 import random
 
-# monkey-patch: get_text_config를 덮어써서 dummy 객체를 반환 (tie_word_embeddings=False)
-PretrainedConfig.get_text_config = lambda self, decoder=True: type("DummyTextConfig", (), {"tie_word_embeddings": False})()
+# 오버라이드: tie_weights()를 빈 함수로 재정의하여 tie_weights 호출을 무시함.
+PreTrainedModel.tie_weights = lambda self: None
 
 torch.set_float32_matmul_precision('high')
 torch.jit.script = lambda f: f
@@ -80,7 +80,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)