Anushree1 commited on
Commit
5ef62cb
Β·
verified Β·
1 Parent(s): 48ba9e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -5,10 +5,13 @@ import numpy as np
5
  from sklearn.metrics.pairwise import cosine_similarity
6
  import os
7
 
8
- # Load speaker embedding model (requires Hugging Face token if private)
9
- model = Inference("pyannote/embedding", window="whole")
10
 
11
- # Load known speaker embeddings
 
 
 
12
  speaker_embeddings = {}
13
  for speaker in os.listdir("known_speakers"):
14
  if speaker.endswith(".wav"):
@@ -29,7 +32,7 @@ def identify_speaker(audio):
29
 
30
  return f"🧍 Identified Speaker: {best_speaker}\nπŸ§ͺ Similarity Score: {best_score:.2f}"
31
 
32
- # Gradio interface
33
  gr.Interface(
34
  fn=identify_speaker,
35
  inputs=gr.Audio(source="microphone", type="filepath", label="πŸŽ™οΈ Upload or record voice"),
 
5
  from sklearn.metrics.pairwise import cosine_similarity
6
  import os
7
 
8
+ # βœ… Use HF token from Hugging Face Space secrets
9
+ hf_token = os.getenv("HF_TOKEN")
10
 
11
+ # πŸ” Load model with authentication
12
+ model = Inference("pyannote/embedding", use_auth_token=hf_token, window="whole")
13
+
14
+ # 🎧 Load known speaker embeddings
15
  speaker_embeddings = {}
16
  for speaker in os.listdir("known_speakers"):
17
  if speaker.endswith(".wav"):
 
32
 
33
  return f"🧍 Identified Speaker: {best_speaker}\nπŸ§ͺ Similarity Score: {best_score:.2f}"
34
 
35
+ # πŸš€ Launch Gradio UI
36
  gr.Interface(
37
  fn=identify_speaker,
38
  inputs=gr.Audio(source="microphone", type="filepath", label="πŸŽ™οΈ Upload or record voice"),