come one again

Dockerfile

@@ -1,25 +1,27 @@
 FROM python:3.9
 
-# Create user
+# Create non-root user
 RUN useradd -m -u 1000 user
 USER user
+
+# Set environment variables
 ENV PATH="/home/user/.local/bin:$PATH"
 ENV NLTK_DATA="/home/user/nltk_data"
 
 # Set working directory
 WORKDIR /app
 
-# Copy requirements and install
+# Copy and install dependencies
 COPY --chown=user requirements.txt .
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Create nltk_data directory and download punkt
+# Prepare NLTK data directory and download punkt
 RUN mkdir -p $NLTK_DATA
 COPY --chown=user download_nltk_data.py .
 RUN python download_nltk_data.py && rm download_nltk_data.py
 
-# Copy the rest of the app
-COPY --chown=user . /app
+# Copy app code
+COPY --chown=user . .
 
-# Start the app
+# Run the app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -7,11 +7,11 @@ import torch
 import nltk
 import os
 
+
 from nltk.tokenize import sent_tokenize
 
-# Download punkt tokenizer if not already present
 nltk_data_path = os.getenv("NLTK_DATA", "/home/user/nltk_data")
-nltk.download("punkt", download_dir=nltk_data_path)
+nltk.data.path.append(nltk_data_path)
 
 app = FastAPI()