Spaces:
Running
on
Zero
Running
on
Zero
Update whisper_cs.py (#23)
Browse files- Update whisper_cs.py (65e012c55da435369e96c03a87f2be53ca725bc5)
Co-authored-by: Sarah Solito <[email protected]>
- 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"
|