beta v1
Browse files
app.py
CHANGED
|
@@ -22,10 +22,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 22 |
model = WhisperModel(MODEL_NAME, device=device, compute_type="float16" if torch.cuda.is_available() else "int8")
|
| 23 |
|
| 24 |
#model_pyannote = Model.from_pretrained("pyannote/speaker-diarization")
|
| 25 |
-
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
|
| 26 |
|
| 27 |
#pipeline = SpeakerDiarization(model_pyannote)
|
| 28 |
|
|
|
|
| 29 |
@spaces.GPU
|
| 30 |
def transcribe(inputs, task):
|
| 31 |
if inputs is None:
|
|
|
|
| 22 |
model = WhisperModel(MODEL_NAME, device=device, compute_type="float16" if torch.cuda.is_available() else "int8")
|
| 23 |
|
| 24 |
#model_pyannote = Model.from_pretrained("pyannote/speaker-diarization")
|
| 25 |
+
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization",os.getenv('HF_TOKEN'))
|
| 26 |
|
| 27 |
#pipeline = SpeakerDiarization(model_pyannote)
|
| 28 |
|
| 29 |
+
|
| 30 |
@spaces.GPU
|
| 31 |
def transcribe(inputs, task):
|
| 32 |
if inputs is None:
|