RayPac006 commited on
Commit
e2e70f5
·
verified ·
1 Parent(s): 1716dc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ compute_type = "float16" if device == "cuda" else "int8" # int8 is faster on CPU
23
 
24
  # 2. Global Model Load (Load once on startup)
25
  print(f"Loading WhisperX model on {device}...")
26
- model = whisperx.load_model("small", device, compute_type=compute_type)
27
 
28
  def generate_lyrics(audio_file_path):
29
  if audio_file_path is None:
 
23
 
24
  # 2. Global Model Load (Load once on startup)
25
  print(f"Loading WhisperX model on {device}...")
26
+ model = whisperx.load_model("medium", device, compute_type=compute_type)
27
 
28
  def generate_lyrics(audio_file_path):
29
  if audio_file_path is None: