diff --git a/README.md b/README.md index b9e4978..c5ceb52 100644 --- a/README.md +++ b/README.md @@ -97,6 +97,10 @@ print(summary_text) ### 💬 Speaker Diarization +```bash +pip install pyannote.audio>=3.1.0 pyannote.core>=5.0.0 pyannote.database>=5.0.1 pyannote.metrics>=3.2.1 pyannote.pipeline>=3.0.1 torchaudio>=2.0.0 +``` + ```python from whisperplus import ( ASRDiarizationPipeline, @@ -122,6 +126,10 @@ print(dialogue) ### ⭐ RAG - Chat with Video(LanceDB) +```bash +pip install sentence-transformers ctransformers langchain +``` + ```python from whisperplus.pipelines.chatbot import ChatWithVideo @@ -140,6 +148,10 @@ print(response) ### 🌠 RAG - Chat with Video(AutoLLM) +```bash +pip install autollm>=0.1.9 +``` + ```python from whisperplus import AutoLLMChatWithVideo @@ -198,7 +210,7 @@ caption(video_path="test.mp4", output_path="output.mp4", language="turkish") ## 😍 Contributing ```bash -pip install -r dev-requirements.txt +pip install pre-commit pre-commit install pre-commit run --all-files ``` diff --git a/requirements.txt b/requirements.txt index dbe2449..6fae836 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,22 +1,8 @@ torch>=2.0.0 torchvision>=0.15.0 -torchaudio>=2.0.0 -gradio>=4.14.0 -langchain>=0.1.0 moviepy>=1.0.3 -numpy>=1.24.1 -pyannote.audio>=3.1.0 -pyannote.core>=5.0.0 -pyannote.database>=5.0.1 -pyannote.metrics>=3.2.1 -pyannote.pipeline>=3.0.1 pytube>=15.0.0 Requests>=2.31.0 -sentence-transformers -ctransformers accelerate -pre-commit==3.4.0 -autollm>=0.1.9 -speechbrain>=0.5.16 bitsandbytes hqq