Update Dockerfile

Dockerfile

@@ -23,26 +23,20 @@ ENV HF_HOME=/cache/huggingface
 ENV UPLOAD_DIR=/tmp/uploads
 ENV NLTK_DATA_DIR=/cache/nltk_data
 ENV PYTHONUNBUFFERED=1
+ENV PYTHONPATH=/app
 
-# Install dependencies
+# Install Python dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt && \
+    pip install --no-cache-dir torchvision
 
 # Pre-download NLTK punkt_tab
 RUN python -c "import nltk; nltk.download('punkt_tab', download_dir='/cache/nltk_data')"
 
-# Pre-download models
-RUN python -c "from transformers import AutoModel, AutoTokenizer, AutoProcessor, AutoModelForQuestionAnswering; \
-    AutoModel.from_pretrained('sshleifer/distilbart-cnn-12-6'); \
-    AutoTokenizer.from_pretrained('sshleifer/distilbart-cnn-12-6'); \
-    AutoModel.from_pretrained('Salesforce/blip-image-captioning-large'); \
-    AutoProcessor.from_pretrained('Salesforce/blip-image-captioning-large'); \
-    AutoModel.from_pretrained('dandelin/vilt-b32-finetuned-vqa'); \
-    AutoProcessor.from_pretrained('dandelin/vilt-b32-finetuned-vqa'); \
-    AutoModelForQuestionAnswering.from_pretrained('distilbert-base-cased-distilled-squad'); \
-    AutoTokenizer.from_pretrained('distilbert-base-cased-distilled-squad'); \
-    AutoModel.from_pretrained('facebook/m2m100_418M'); \
-    AutoTokenizer.from_pretrained('facebook/m2m100_418M')"
+# Copy model download script
+COPY download_models.py .
+# Run model download with error handling
+RUN python download_models.py
 
 # Copy application
 COPY . .