Make HF_TOKEN optional for pyannote models

parent 45156b52
......@@ -27,7 +27,7 @@ This Python application transcribes audio files with speaker diarization and tim
```bash
pip install -r requirements.txt
```
5. Set Hugging Face token:
5. Optional: Set Hugging Face token for pyannote models (required for first download):
```bash
export HF_TOKEN=your_huggingface_token
```
......
......@@ -27,12 +27,8 @@ def main():
processor = AutoProcessor.from_pretrained("Qwen/Qwen-Omni-7B")
# Load diarization pipeline
# Note: Requires Hugging Face token set as HF_TOKEN environment variable
hf_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)
# Optional: Set HF_TOKEN environment variable for pyannote authentication (required for first download)
diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=os.getenv("HF_TOKEN"))
# Load audio
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