Spaces:
Running
Running
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,
|