Make HF_TOKEN optional for pyannote models

parent 45156b52
...@@ -27,7 +27,7 @@ This Python application transcribes audio files with speaker diarization and tim ...@@ -27,7 +27,7 @@ This Python application transcribes audio files with speaker diarization and tim
```bash ```bash
pip install -r requirements.txt pip install -r requirements.txt
``` ```
5. Set Hugging Face token: 5. Optional: Set Hugging Face token for pyannote models (required for first download):
```bash ```bash
export HF_TOKEN=your_huggingface_token export HF_TOKEN=your_huggingface_token
``` ```
......
...@@ -27,12 +27,8 @@ def main(): ...@@ -27,12 +27,8 @@ def main():
processor = AutoProcessor.from_pretrained("Qwen/Qwen-Omni-7B") processor = AutoProcessor.from_pretrained("Qwen/Qwen-Omni-7B")
# Load diarization pipeline # Load diarization pipeline
# Note: Requires Hugging Face token set as HF_TOKEN environment variable # Optional: Set HF_TOKEN environment variable for pyannote authentication (required for first download)
hf_token = os.getenv("HF_TOKEN") diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=os.getenv("HF_TOKEN"))
if not hf_token:
print("Error: Please set HF_TOKEN environment variable for pyannote authentication.")
return
diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=hf_token)
# Load audio # Load audio
audio, sr = librosa.load(audio_file, sr=16000) audio, sr = librosa.load(audio_file, sr=16000)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment