Update Dockerfile

Dockerfile

@@ -1,7 +1,9 @@
 FROM huggingface/autotrain-advanced:latest
 
 RUN pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced \
-    && pip install nltk  \
-    && python -m nltk.downloader -d /app/nltk_data punkt \
+    && pip install nltk \
+    && python -c "import nltk; nltk.download('punkt', download_dir='/app/nltk_data')"
 
-CMD autotrain app --host 0.0.0.0 --port 7860 --workers 1
+ENV NLTK_DATA="/app/nltk_data"
+
+CMD ["autotrain", "app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]