Spaces:
Sleeping
Sleeping
ShobhitKori
commited on
Commit
·
62522ae
1
Parent(s):
e94ab80
Updated codet5_model.py
Browse files- whisper_model.py +2 -1
whisper_model.py
CHANGED
|
@@ -5,7 +5,8 @@ import logging
|
|
| 5 |
# Add FFmpeg directory to PATH
|
| 6 |
# os.environ["PATH"] += os.pathsep + r"C:\ffmpeg\bin"
|
| 7 |
|
| 8 |
-
WHISPER_MODEL_DIR = "
|
|
|
|
| 9 |
|
| 10 |
model = whisper.load_model("small", download_root=WHISPER_MODEL_DIR)
|
| 11 |
|
|
|
|
| 5 |
# Add FFmpeg directory to PATH
|
| 6 |
# os.environ["PATH"] += os.pathsep + r"C:\ffmpeg\bin"
|
| 7 |
|
| 8 |
+
WHISPER_MODEL_DIR = "./data/whisper-small"
|
| 9 |
+
os.makedirs(WHISPER_MODEL_DIR, exist_ok=True)
|
| 10 |
|
| 11 |
model = whisper.load_model("small", download_root=WHISPER_MODEL_DIR)
|
| 12 |
|