Spaces:
Sleeping
Sleeping
jhj0517
commited on
Commit
·
988860f
1
Parent(s):
e862b08
Fix factory function
Browse files
modules/whisper/whisper_factory.py
CHANGED
|
@@ -59,14 +59,14 @@ class WhisperFactory:
|
|
| 59 |
diarization_model_dir=diarization_model_dir,
|
| 60 |
uvr_model_dir=uvr_model_dir
|
| 61 |
)
|
| 62 |
-
elif whisper_type
|
| 63 |
return WhisperInference(
|
| 64 |
model_dir=whisper_model_dir,
|
| 65 |
output_dir=output_dir,
|
| 66 |
diarization_model_dir=diarization_model_dir,
|
| 67 |
uvr_model_dir=uvr_model_dir
|
| 68 |
)
|
| 69 |
-
elif whisper_type
|
| 70 |
return InsanelyFastWhisperInference(
|
| 71 |
model_dir=insanely_fast_whisper_model_dir,
|
| 72 |
output_dir=output_dir,
|
|
|
|
| 59 |
diarization_model_dir=diarization_model_dir,
|
| 60 |
uvr_model_dir=uvr_model_dir
|
| 61 |
)
|
| 62 |
+
elif whisper_type == WhisperImpl.WHISPER:
|
| 63 |
return WhisperInference(
|
| 64 |
model_dir=whisper_model_dir,
|
| 65 |
output_dir=output_dir,
|
| 66 |
diarization_model_dir=diarization_model_dir,
|
| 67 |
uvr_model_dir=uvr_model_dir
|
| 68 |
)
|
| 69 |
+
elif whisper_type == WhisperImpl.INSANELY_FAST_WHISPER:
|
| 70 |
return InsanelyFastWhisperInference(
|
| 71 |
model_dir=insanely_fast_whisper_model_dir,
|
| 72 |
output_dir=output_dir,
|