chompionsawelo commited on
Commit
8caee60
·
1 Parent(s): 14a5c51

adjust token

Browse files
Files changed (2) hide show
  1. app.py +0 -3
  2. diarization.py +4 -1
app.py CHANGED
@@ -4,9 +4,6 @@ import ffmpeg
4
  import gradio as gr
5
  import os
6
 
7
- hugging_face_token = os.environ["HUGGING_FACE_TOKEN"]
8
- login(token=hugging_face_token)
9
-
10
  # Get video & convert it
11
  inputs_interface = gr.inputs.Video(label="Insert video")
12
  in_file = ffmpeg.input(inputs_interface).audio
 
4
  import gradio as gr
5
  import os
6
 
 
 
 
7
  # Get video & convert it
8
  inputs_interface = gr.inputs.Video(label="Insert video")
9
  in_file = ffmpeg.input(inputs_interface).audio
diarization.py CHANGED
@@ -1,9 +1,12 @@
1
  from pyannote.audio import Pipeline
2
  from pydub import AudioSegment
 
3
  import torch
4
  import json
5
 
6
- pipeline = Pipeline.from_pretrained('pyannote/speaker-diarization')
 
 
7
  device = torch.device("cpu")
8
  pipeline.to(device)
9
 
 
1
  from pyannote.audio import Pipeline
2
  from pydub import AudioSegment
3
+ import os
4
  import torch
5
  import json
6
 
7
+ hugging_face_token = os.environ["HUGGING_FACE_TOKEN"]
8
+ pipeline = Pipeline.from_pretrained(
9
+ 'pyannote/speaker-diarization', use_auth_token=hugging_face_token)
10
  device = torch.device("cpu")
11
  pipeline.to(device)
12