fix: errors

Dockerfile

@@ -22,6 +22,9 @@ RUN pip install --no-cache-dir -r requirements.txt
 RUN pip install packaging ninja
 RUN pip install flash-attn --no-build-isolation
 
+# Install additional CUDA dependencies for Flash Attention 2
+RUN pip install triton
+
 # Expose port 7860 for Gradio
 EXPOSE 7860
 

requirements.txt

@@ -1,15 +1,14 @@
-huggingface_hub==0.22.2
-transformers
-datasets
-pillow
-numpy
 torch
 gradio
-streaming-stt-nemo
+transformers
+huggingface_hub
 edge-tts
-asyncio
-spaces
-torchvision
-accelerate
+nemo_toolkit
+requests
 beautifulsoup4
-requests
+Pillow
+spaces
+datasets
+packaging
+ninja
+triton