diff --git a/Dockerfile b/Dockerfile index f12b154..9a8398e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -15,12 +15,14 @@ RUN apt-get update && apt-get install -y \ # Copy requirements first for better Docker layer caching COPY requirements.txt . -# Install Python dependencies with pinned versions -RUN pip install --no-cache-dir -r requirements.txt +# Upgrade pip and install build tools +RUN pip install --upgrade pip setuptools wheel -# Optional: Install CUDA-specific PyTorch if GPU support needed -# Uncomment and modify for your CUDA version: -# RUN pip install --force-reinstall torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0+cu118 --index-url https://download.pytorch.org/whl/cu118 +# Install PyTorch CPU version first (for non-GPU builds) +RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu + +# Install remaining Python dependencies +RUN pip install --no-cache-dir -r requirements.txt # Copy application code COPY . . diff --git a/requirements.txt b/requirements.txt index 13da6dc..6b28359 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ # OBS Recording Transcriber Dependencies # Core dependencies -streamlit>=1.26.0 +# streamlit>=1.30.0 required for protobuf>=5.0 compatibility +streamlit>=1.30.0 moviepy>=1.0.3 openai-whisper>=20231117 requests>=2.28.0 @@ -28,7 +29,8 @@ pytorch-lightning>=2.0.0 # Other dependencies iso639>=0.1.4 -protobuf>=3.20.0,<5.0.0 +# protobuf>=5.0 required by opentelemetry (pyannote.audio dependency) +protobuf>=5.0.0 matplotlib>=3.5.0 soundfile>=0.10.3 ffmpeg-python>=0.2.0