Update Dockerfile

Dockerfile

@@ -1,8 +1,10 @@
 FROM python:3.10-slim
 
 ENV PYTHONUNBUFFERED=1
-ENV TRANSFORMERS_CACHE=/app/cache
+# Replace TRANSFORMERS_CACHE with HF_HOME
+ENV HF_HOME=/app/cache
 ENV UPLOAD_DIR=/app/uploads
+ENV NLTK_DATA=/tmp/nltk_data
 ENV PORT=7860
 
 # Install system dependencies
@@ -16,8 +18,8 @@ RUN apt-get update && \
     && rm -rf /var/lib/apt/lists/*
 
 # Create directories and set permissions
-RUN mkdir -p /app/cache /app/uploads /app/static /app/images && \
-    chmod -R 777 /app
+RUN mkdir -p /app/cache /app/uploads /app/static /app/images /tmp/nltk_data && \
+    chmod -R 777 /app /tmp/nltk_data
 
 WORKDIR /app
 
@@ -26,6 +28,10 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -U pip setuptools wheel && \
     pip install --no-cache-dir -r requirements.txt
 
+# Clear cache to avoid conflicts with model names
+RUN rm -rf /app/cache/* && \
+    chmod -R 777 /app/cache
+
 # Copy application code
 COPY . .
 
@@ -37,4 +43,4 @@ HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
 # Run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]