Raí Santos commited on
Commit ·
3af6bcf
1
Parent(s): 7c68fa1
feat: Complete optimization with 3 bugs fixed + backend-only
Browse files- Dockerfile +3 -0
- backend/requirements.txt +2 -5
Dockerfile
CHANGED
|
@@ -27,6 +27,9 @@ RUN chmod -R 777 /app
|
|
| 27 |
# Upgrade pip and install build tools
|
| 28 |
RUN pip install --no-cache-dir --upgrade pip setuptools wheel
|
| 29 |
|
|
|
|
|
|
|
|
|
|
| 30 |
# Install backend dependencies
|
| 31 |
COPY backend/requirements.txt .
|
| 32 |
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
| 27 |
# Upgrade pip and install build tools
|
| 28 |
RUN pip install --no-cache-dir --upgrade pip setuptools wheel
|
| 29 |
|
| 30 |
+
# Install Torch CPU specifically (Prevents Erro 500 and OOM)
|
| 31 |
+
RUN pip install --no-cache-dir torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cpu
|
| 32 |
+
|
| 33 |
# Install backend dependencies
|
| 34 |
COPY backend/requirements.txt .
|
| 35 |
RUN pip install --no-cache-dir -r requirements.txt
|
backend/requirements.txt
CHANGED
|
@@ -1,14 +1,11 @@
|
|
| 1 |
-
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 2 |
fastapi
|
| 3 |
uvicorn
|
| 4 |
python-multipart
|
| 5 |
-
torch==2.5.1
|
| 6 |
-
torchaudio==2.5.1
|
| 7 |
-
git+https://github.com/m-bain/whisperX.git@v3.1.1
|
| 8 |
-
pyannote.audio==3.3.1
|
| 9 |
python-docx
|
| 10 |
transformers
|
| 11 |
pandas==2.2.2
|
| 12 |
onnxruntime
|
| 13 |
accelerate
|
| 14 |
nest_asyncio
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
fastapi
|
| 2 |
uvicorn
|
| 3 |
python-multipart
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
python-docx
|
| 5 |
transformers
|
| 6 |
pandas==2.2.2
|
| 7 |
onnxruntime
|
| 8 |
accelerate
|
| 9 |
nest_asyncio
|
| 10 |
+
pyannote.audio==3.3.1
|
| 11 |
+
git+https://github.com/m-bain/whisperX.git@v3.1.1
|