palli23 commited on
Commit
75fe8a3
·
verified ·
1 Parent(s): e4aa950

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -1,9 +1,11 @@
1
- torch==2.2.0+cu121
2
- torchaudio==2.2.0+cu121
3
  --extra-index-url https://download.pytorch.org/whl/cu121
 
 
4
 
5
  transformers==4.44.2
6
  gradio==4.44.0
7
  huggingface-hub==0.24.6
8
- git+https://github.com/m-bain/whisperX.git # WhisperX main (latest 2025)
9
- numpy==2.1.1 # Fixes NumPy conflicts
 
 
1
+ # Pin PyTorch for ZeroGPU (fixes build + CUDA)
 
2
  --extra-index-url https://download.pytorch.org/whl/cu121
3
+ torch==2.2.0
4
+ torchaudio==2.2.0
5
 
6
  transformers==4.44.2
7
  gradio==4.44.0
8
  huggingface-hub==0.24.6
9
+ kenlm==0.2.0
10
+ pyannote-audio==3.1.1
11
+ numpy==1.26.4 # Compatible with Torch 2.2.0 (avoids NumPy crash)