Upload Dockerfile

Dockerfile

@@ -3,18 +3,23 @@ FROM python:3.8
 # Install required system dependencies
 RUN apt-get update && apt-get install -y libopenblas-dev git curl && rm -rf /var/lib/apt/lists/*
 
+# Set environment variables for cache
+ENV TRANSFORMERS_CACHE=/app/cache/huggingface
+ENV NUMBA_CACHE_DIR=/app/cache/numba
+
 # Set the working directory inside the container
 WORKDIR /app
 
+# Create necessary directories
+RUN mkdir -p /app/modelsBioembed /app/models_folder /app/Samples \
+    && mkdir -p $TRANSFORMERS_CACHE $NUMBA_CACHE_DIR
+
 # Copy the requirements file into the container
 COPY requirements.txt .
 
 # Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Create necessary directories (but don't download models here!)
-RUN mkdir -p /app/modelsBioembed /app/models_folder /app/Samples
-
 # Copy the entire project to the container
 COPY . .
 
@@ -25,6 +30,34 @@ EXPOSE 7860
 CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]
 
 
+
+# FROM python:3.8
+
+# # Install required system dependencies
+# RUN apt-get update && apt-get install -y libopenblas-dev git curl && rm -rf /var/lib/apt/lists/*
+
+# # Set the working directory inside the container
+# WORKDIR /app
+
+# # Copy the requirements file into the container
+# COPY requirements.txt .
+
+# # Install dependencies
+# RUN pip install --no-cache-dir -r requirements.txt
+
+# # Create necessary directories (but don't download models here!)
+# RUN mkdir -p /app/modelsBioembed /app/models_folder /app/Samples
+
+# # Copy the entire project to the container
+# COPY . .
+
+# # Expose the port for Flask
+# EXPOSE 7860
+
+# # Run the app with Gunicorn
+# CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]
+
+
 # # Use Python 3.8 as the base image
 # FROM python:3.8