Spaces:
Running
on
Zero
Running
on
Zero
Update whisper_cs.py
#23
by
ssolito
- opened
- whisper_cs.py +1 -1
whisper_cs.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from pydub import AudioSegment
|
| 2 |
import os
|
| 3 |
import torchaudio
|
|
@@ -7,7 +8,6 @@ from transformers import pipeline, WhisperForConditionalGeneration, WhisperProce
|
|
| 7 |
from pyannote.audio import Pipeline as DiarizationPipeline
|
| 8 |
import whisperx
|
| 9 |
import whisper_timestamped as whisper_ts
|
| 10 |
-
import spaces
|
| 11 |
from typing import Dict
|
| 12 |
|
| 13 |
device = 0 if torch.cuda.is_available() else "cpu"
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
from pydub import AudioSegment
|
| 3 |
import os
|
| 4 |
import torchaudio
|
|
|
|
| 8 |
from pyannote.audio import Pipeline as DiarizationPipeline
|
| 9 |
import whisperx
|
| 10 |
import whisper_timestamped as whisper_ts
|
|
|
|
| 11 |
from typing import Dict
|
| 12 |
|
| 13 |
device = 0 if torch.cuda.is_available() else "cpu"
|