Upload 3 files

Dockerfile

@@ -15,7 +15,8 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt && \
-    patchelf --clear-execstack /usr/local/lib/python3.10/site-packages/onnxruntime/capi/onnxruntime_pybind11_state.cpython-310-x86_64-linux-gnu.so
+    patchelf --clear-execstack /usr/local/lib/python3.10/site-packages/onnxruntime/capi/onnxruntime_pybind11_state.cpython-310-x86_64-linux-gnu.so && \
+    find /usr/local/lib/python3.10/site-packages/tokenizers -name "*.so" -exec patchelf --clear-execstack {} +
 
 COPY . .
 

app.py

@@ -4,37 +4,12 @@ from pix2text import Pix2Text
 import io
 from PIL import Image
 import uvicorn
+import os
 
-app = FastAPI()
-
-# CORSを許可
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-import transformers
-from transformers import XLMRobertaTokenizer, XLMRobertaTokenizerFast
-
-# 高速トークナイザーのクラス自体を低速版で上書き
-transformers.models.xlm_roberta.tokenization_xlm_roberta_fast.XLMRobertaTokenizerFast = XLMRobertaTokenizer
-transformers.XLMRobertaTokenizerFast = XLMRobertaTokenizer
+# 高速トークナイザーの並列処理によるデッドロックを回避
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-# from_pretrained において use_fast=False を強制し、None エラーを回避するパッチ
-original_from_pretrained = XLMRobertaTokenizer.from_pretrained
-
-@classmethod
-def patched_from_pretrained(cls, *args, **kwargs):
-    kwargs['use_fast'] = False
-    # XLMRobertaTokenizer (Slow) の from_pretrained を呼ぶ
-    return original_from_pretrained(*args, **kwargs)
-
-# 両方のクラスの from_pretrained を差し替え
-XLMRobertaTokenizer.from_pretrained = patched_from_pretrained
-XLMRobertaTokenizerFast.from_pretrained = patched_from_pretrained
+app = FastAPI()
 
 # Pix2Textを初期化
 p2t = Pix2Text()

requirements.txt

@@ -8,7 +8,6 @@ torchvision==0.18.1+cpu
 --extra-index-url https://download.pytorch.org/whl/cpu
 transformers==4.40.2
 tokenizers==0.19.1
-sentencepiece
 huggingface-hub==0.24.5
 numpy<2.0.0
 onnxruntime==1.19.0