Update Dockerfile

Dockerfile

@@ -13,8 +13,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Copy the requirements file
 COPY requirements.txt requirements.txt
 
-# Install Python packages
-RUN pip install --no-cache-dir -r requirements.txt
+# Install Python packages with timeout increase
+RUN pip install --no-cache-dir --timeout=1000 -r requirements.txt
 
 # Copy application code
 COPY . /app
@@ -28,13 +28,18 @@ RUN chown -R user:user /app
 # Switch to the non-root user
 USER user
 
-# Expose the port Gunicorn will run on (Using 7860 as in CMD)
+# Expose the port
 EXPOSE 7860
 
-# Set environment variables for Flask host and port
+# Set environment variables
 ENV FLASK_HOST=0.0.0.0
 ENV FLASK_PORT=7860
-# ENV FLASK_DEBUG="False" # Explicitly set if needed, app.py defaults to False if var not present
+ENV FLASK_DEBUG=False
 
-# Command to run the app (app.py will now pick up FLASK_HOST and FLASK_PORT from ENV)
+# CRITICAL: Set HF-specific env vars
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV HF_HOME=/tmp/hf_home
+ENV TORCH_HOME=/tmp/torch_home
+
+# Command to run the app
 CMD ["python", "app.py"]