Update Dockerfile

Dockerfile

@@ -1,32 +1,35 @@
-FROM python:3.10-slim
-
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONUNBUFFERED=1
-ENV DEBIAN_FRONTEND=noninteractive
-ENV MODEL_PATH=RufusRubin777/GOT-OCR2_0_CPU
-
-WORKDIR /app
-
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    libpq-dev \
-    && rm -rf /var/lib/apt/lists/*
-
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-RUN mkdir -p /root/.cache/huggingface
-VOLUME /root/.cache/huggingface
-
-RUN python -c "from transformers_modules.RufusRubin777.GOT_OCR2_0_CPU.modeling_GOT import GOTModel, GOTConfig; \
-    from transformers import AutoTokenizer; \
-    model_path='RufusRubin777/GOT-OCR2_0_CPU'; \
-    config = GOTConfig.from_pretrained(model_path); \
-    model = GOTModel.from_pretrained(model_path, config=config); \
-    tokenizer = AutoTokenizer.from_pretrained(model_path)"
-
-COPY . .
-
-EXPOSE 7860
-
+FROM python:3.10-slim
+
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive
+ENV MODEL_PATH=RufusRubin777/GOT-OCR2_0_CPU
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    libpq-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Download NLTK data
+RUN python -m nltk.downloader punkt punkt_tab
+
+RUN mkdir -p /root/.cache/huggingface
+VOLUME /root/.cache/huggingface
+
+RUN python -c "from transformers_modules.stepfun-ai.GOT-OCR2_0.modeling_GOT import GOTModel, GOTConfig; \
+    from transformers import AutoTokenizer; \
+    model_path='stepfun-ai/GOT-OCR2_0'; \
+    config = GOTConfig.from_pretrained(model_path); \
+    model = GOTModel.from_pretrained(model_path, config=config); \
+    tokenizer = AutoTokenizer.from_pretrained(model_path)"
+
+COPY . .
+
+EXPOSE 7860
+
 CMD ["python", "app.py"]