Spaces:
Running
Running
jhj0517
commited on
Commit
·
0184fe2
1
Parent(s):
54a7493
Fix gradio input visibility by implementation type
Browse files
modules/whisper/data_classes.py
CHANGED
|
@@ -297,12 +297,12 @@ class WhisperParams(BaseParams):
|
|
| 297 |
def to_gradio_inputs(cls,
|
| 298 |
defaults: Optional[Dict] = None,
|
| 299 |
only_advanced: Optional[bool] = True,
|
| 300 |
-
whisper_type: Optional[
|
| 301 |
available_models: Optional[List] = None,
|
| 302 |
available_langs: Optional[List] = None,
|
| 303 |
available_compute_types: Optional[List] = None,
|
| 304 |
compute_type: Optional[str] = None):
|
| 305 |
-
whisper_type = WhisperImpl.FASTER_WHISPER if whisper_type is None else whisper_type
|
| 306 |
|
| 307 |
inputs = []
|
| 308 |
if not only_advanced:
|
|
@@ -491,13 +491,13 @@ class WhisperParams(BaseParams):
|
|
| 491 |
)
|
| 492 |
]
|
| 493 |
|
| 494 |
-
if whisper_type
|
| 495 |
for input_component in faster_whisper_inputs:
|
| 496 |
-
input_component.visible =
|
| 497 |
|
| 498 |
-
if whisper_type
|
| 499 |
for input_component in insanely_fast_whisper_inputs:
|
| 500 |
-
input_component.visible =
|
| 501 |
|
| 502 |
inputs += faster_whisper_inputs + insanely_fast_whisper_inputs
|
| 503 |
|
|
|
|
| 297 |
def to_gradio_inputs(cls,
|
| 298 |
defaults: Optional[Dict] = None,
|
| 299 |
only_advanced: Optional[bool] = True,
|
| 300 |
+
whisper_type: Optional[str] = None,
|
| 301 |
available_models: Optional[List] = None,
|
| 302 |
available_langs: Optional[List] = None,
|
| 303 |
available_compute_types: Optional[List] = None,
|
| 304 |
compute_type: Optional[str] = None):
|
| 305 |
+
whisper_type = WhisperImpl.FASTER_WHISPER.value if whisper_type is None else whisper_type.strip().lower()
|
| 306 |
|
| 307 |
inputs = []
|
| 308 |
if not only_advanced:
|
|
|
|
| 491 |
)
|
| 492 |
]
|
| 493 |
|
| 494 |
+
if whisper_type != WhisperImpl.FASTER_WHISPER.value:
|
| 495 |
for input_component in faster_whisper_inputs:
|
| 496 |
+
input_component.visible = False
|
| 497 |
|
| 498 |
+
if whisper_type != WhisperImpl.INSANELY_FAST_WHISPER.value:
|
| 499 |
for input_component in insanely_fast_whisper_inputs:
|
| 500 |
+
input_component.visible = False
|
| 501 |
|
| 502 |
inputs += faster_whisper_inputs + insanely_fast_whisper_inputs
|
| 503 |
|