pep 8 and change words and links

Author: jianfch

.github/workflows/test.yml

@@ -85,4 +85,4 @@ jobs:
       - name: Run MLX align tests
         run: python test/test_align.py load_mlx_whisper
       - name: Run MLX refine tests
-        run: python test/test_refine.py load_mlx_whisper
+        run: python test/test_refine.py load_mlx_whisper

README.md

@@ -537,9 +537,9 @@ stable-ts audio.mp3 -o audio.srt -hw
 </details>
 
 <details>
-<summary>MLX Whisper for Apple GPU's</summary>
+<summary>MLX Whisper (on Apple Silicon)</summary>
 
-Transcribe faster on Apple devices with [MLX Whisper](https://huggingface.co/mlx-community/whisper-large-v3-turbo):
+Transcribe faster on Apple devices with [MLX Whisper](https://github.com/ml-explore/mlx-examples/tree/main/whisper):
 ```
 pip install -U stable-ts[mlx]
 ```
@@ -550,10 +550,6 @@ import stable_whisper
 model = stable_whisper.load_mlx_whisper('base')
 result = model.transcribe('audio.mp3')
 ```
-Supports the [various versions on Hugging Face](https://huggingface.co/models?sort=downloads&search=mlx-community%2Fwhisper&sort=downloads):
-```python
-model = stable_whisper.load_hf_whisper('mlx-community/whisper-base.en-mlx')
-```
 
 
 <details>

stable_whisper/alignment.py

@@ -423,10 +423,13 @@ def compute_timestamps(audio_segment: torch.Tensor, word_tokens: List[WordToken]
             )
             return [w for seg in temp_segments for w in seg['words']]
 
-
     elif model_type == 'mlx':
-        from mlx_whisper.audio import N_FRAMES as MLX_N_FRAMES, SAMPLE_RATE as MLX_SAMPLE_RATE, log_mel_spectrogram as log_mel_spectrogram_mx, \
+        from mlx_whisper.audio import (
+            N_FRAMES as MLX_N_FRAMES,
+            SAMPLE_RATE as MLX_SAMPLE_RATE,
+            log_mel_spectrogram as log_mel_spectrogram_mx,
             pad_or_trim as pad_or_trim_mx
+        )
         import mlx.core as mx
         import mlx_whisper.timing as timing
 
@@ -664,8 +667,11 @@ def inference_func(audio_segment: torch, tokens: List[int]) -> torch.Tensor:
             return token_probs
 
     elif model_type == 'mlx':
-        from mlx_whisper.audio import N_FRAMES_MLX, log_mel_spectrogram as log_mel_spectrogram_mx, \
+        from mlx_whisper.audio import (
+            N_FRAMES_MLX,
+            log_mel_spectrogram as log_mel_spectrogram_mx,
             pad_or_trim as pad_or_trim_mx
+        )
         import mlx.core as mx
 
         def inference_func(audio_batch_torch: torch.Tensor, tokens: List[int]) -> torch.Tensor:

stable_whisper/whisper_word_level/cli.py

@@ -374,7 +374,8 @@ def url_to_path(url: str):
                              '(https://huggingface.co/openai/whisper-large-v3); '
                              'note: some features may not be available')
     parser.add_argument('--mlx_whisper', '-mlx', action='store_true',
-                        help='whether to use mlx-whisper (https://huggingface.co/mlx-community/whisper-large-v3-turbo); '
+                        help='whether to use mlx-whisper '
+                             '(https://github.com/ml-explore/mlx-examples/tree/main/whisper); '
                              'note: some features may not be available')
 
     parser.add_argument('--persist', '-p', action='store_true',

stable_whisper/whisper_word_level/mlx_whisper.py

@@ -331,4 +331,4 @@ def load_mlx_whisper(model_name: str, dtype=None, **model_kwargs):
     model.align_words = MethodType(align_words, model)
     model.refine = MethodType(refine, model)
 
-    return model
+    return model