Update app.py

app.py

@@ -3,11 +3,33 @@ import os
 os.environ['CUDA_VISIBLE_DEVICES'] = ''
 os.environ['ZEROGPU'] = '1'  # ZeroGPU 환경임을 표시
 
-# Safetensors 사용 강제 및 torch 버전 체크 우회
+# Safetensors 사용 강제
 os.environ['SAFETENSORS_FAST_GPU'] = '1'
 os.environ['TRANSFORMERS_OFFLINE'] = '0'
 os.environ['TRANSFORMERS_USE_SAFETENSORS'] = '1'
-os.environ['DISABLE_TORCH_LOAD_SAFETY_CHECK'] = '1'  # This might help bypass the check
+
+# Patch transformers to add missing SiglipImageProcessor
+import sys
+from types import ModuleType
+
+# Create mock for SiglipImageProcessor before importing transformers
+if 'transformers' not in sys.modules:
+    # Create a dummy SiglipImageProcessor class
+    class DummySiglipImageProcessor:
+        def __init__(self, *args, **kwargs):
+            pass
+    
+    class DummySiglipVisionModel:
+        def __init__(self, *args, **kwargs):
+            pass
+    
+    # Pre-patch transformers module
+    transformers_module = ModuleType('transformers')
+    transformers_module.SiglipImageProcessor = DummySiglipImageProcessor
+    transformers_module.SiglipVisionModel = DummySiglipVisionModel
+    
+    # Add to sys.modules
+    sys.modules['transformers'] = transformers_module
 
 import spaces  # spaces import는 환경 설정 후에
 import shlex
@@ -16,17 +38,9 @@ import subprocess
 # 라이브러리 버전 호환성 문제 해결
 subprocess.run(shlex.split("pip install pip==24.0"), check=True)
 
-# Safetensors를 기본으로 사용하도록 transformers 설정
+# Safetensors 설치
 subprocess.run(shlex.split("pip install safetensors --upgrade"), check=True)
 
-# Monkey patch to force safetensors loading
-import sys
-if 'transformers' in sys.modules:
-    del sys.modules['transformers']
-    
-# transformers import 전에 환경 변수 설정
-os.environ['TRANSFORMERS_USE_SAFETENSORS'] = '1'
-
 subprocess.run(
     shlex.split(
         "pip install package/onnxruntime_gpu-1.17.0-cp310-cp310-manylinux_2_28_x86_64.whl --force-reinstall --no-deps"
@@ -38,6 +52,40 @@ subprocess.run(
     ), check=True
 )
 
+# Import transformers and patch it
+import transformers
+
+# Add missing classes if they don't exist
+if not hasattr(transformers, 'SiglipImageProcessor'):
+    class SiglipImageProcessor:
+        def __init__(self, *args, **kwargs):
+            # Fallback to CLIPImageProcessor
+            self._processor = transformers.CLIPImageProcessor(*args, **kwargs)
+        
+        def __getattr__(self, name):
+            return getattr(self._processor, name)
+    
+    transformers.SiglipImageProcessor = SiglipImageProcessor
+
+if not hasattr(transformers, 'SiglipVisionModel'):
+    class SiglipVisionModel:
+        def __init__(self, *args, **kwargs):
+            # Fallback to CLIPVisionModel
+            from transformers import CLIPVisionModel
+            self._model = CLIPVisionModel(*args, **kwargs)
+        
+        def __getattr__(self, name):
+            return getattr(self._model, name)
+        
+        @classmethod
+        def from_pretrained(cls, *args, **kwargs):
+            instance = cls.__new__(cls)
+            from transformers import CLIPVisionModel
+            instance._model = CLIPVisionModel.from_pretrained(*args, **kwargs)
+            return instance
+    
+    transformers.SiglipVisionModel = SiglipVisionModel
+
 # 모델 체크포인트 다운로드 및 torch 설정
 if __name__ == "__main__":
     from huggingface_hub import snapshot_download