app.py

Dockerfile

@@ -1,24 +1,40 @@
-# Use an official Python runtime as a parent image
-FROM python:3.9-slim
+FROM python:3.10-slim
 
-# Set the working directory in the container
 WORKDIR /code
 
-# Copy the requirements file into the container at /code
-COPY requirements.txt .
+# Install system dependencies required for building some Python packages
+RUN apt-get update && apt-get install -y \
+    gcc \
+    g++ \
+    git \
+    wget \
+    ffmpeg \
+    && rm -rf /var/lib/apt/lists/*
 
-# Install any needed packages specified in requirements.txt
-# --no-cache-dir: Disables the cache which reduces the image size
-# --trusted-host pypi.python.org: Solves issues with SSL/TLS verification in some environments
-RUN pip install --no-cache-dir --trusted-host pypi.python.org -r requirements.txt
+# Set environment variables for Hugging Face cache and tokenizers
+ENV HF_HOME=/code/cache
+ENV TRANSFORMERS_CACHE=/code/cache
+ENV TOKENIZERS_PARALLELISM=false
 
-# Copy the rest of the application files into the container at /code
-COPY . .
+# Copy requirements first for better Docker layer caching
+COPY requirements.txt /code/requirements.txt
 
-# Make port 7860 available to the world outside this container
+# Upgrade pip and install Python dependencies
+RUN pip install --no-cache-dir --upgrade pip
+RUN pip install --no-cache-dir -r /code/requirements.txt
+
+# Pre-download the models at build time to avoid runtime delays.
+# This ensures the application starts up quickly in the Space.
+RUN python -c "from transformers import pipeline; import whisper; pipeline('summarization', model='facebook/bart-large-cnn'); whisper.load_model('base')"
+
+# Copy the rest of the application code into the container
+COPY . /code/
+
+# Create the cache directory (the ENV variables point here)
+RUN mkdir -p /code/cache
+
+# Expose the port Gradio runs on
 EXPOSE 7860
 
-# Define the command to run the application
-# Use the --server-name 0.0.0.0 flag to make the app accessible from outside the container
-# Use the --share flag to create a public link if needed (optional)
+# Command to run the application
 CMD ["python3", "app.py"]

requirements.txt

@@ -1,4 +1,4 @@
-gradio==4.44.1
+gradio==4.44.0
 transformers==4.35.2
 torch==2.1.1
 openai-whisper==20231117
@@ -7,4 +7,5 @@ python-docx==1.1.0
 datasets==2.14.6
 accelerate==0.24.1
 sentencepiece==0.1.99
-protobuf==4.25.0
+protobuf==4.25.0
+numpy<2.0