Update Dockerfile

Dockerfile

@@ -1,6 +1,10 @@
 # Use official Python 3.10 slim image for a smaller footprint
 FROM python:3.10-slim
 
+# Set environment variable for Hugging Face cache
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
+
 # Set working directory
 WORKDIR /app
 
@@ -10,14 +14,16 @@ COPY requirements.txt .
 # Install system dependencies and Python packages
 RUN apt-get update && apt-get install -y \
     gcc \
+    libgomp1 \  # Required for PyTorch
     && rm -rf /var/lib/apt/lists/* \
     && pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY app.py .
 
-# Create cache directory for Hugging Face models
-RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+# Create cache and logs directories with write permissions
+RUN mkdir -p /app/cache /app/logs \
+    && chmod -R 777 /app/cache /app/logs
 
 # Expose port for FastAPI
 EXPOSE 8000