Switch to distil large

app.py

@@ -3,21 +3,17 @@ import spaces
 import torch
 import io
 import os.path
-import urllib.request
+from huggingface_hub import hf_hub_download
 import whisper
 
-os.makedirs("/home/user/.cache/whisper")
-urllib.request.urlretrieve(
-    "https://openaipublic.azureedge.net/main/whisper/models/f953ad0fd29cacd07d5a9eda5624af0f6bcf2258be67c92b79389873d91e0872/small.en.pt",
-    "/home/user/.cache/whisper/small.en.pt"
-)
+model_path = hf_hub_download(repo_id="distil-whisper/distil-large-v3-openai", filename="model.bin")
 
 writer = whisper.utils.get_writer("srt", "/dev/null")
 
 @spaces.GPU
 def generate(file, progress=gr.Progress(track_tqdm=True)):
     # get file to type bytes somehow
-    model = whisper.load_model("small.en", device="cuda")
+    model = whisper.load_model(model_path, device="cuda")
     audio = whisper.load_audio(file)
     result = model.transcribe(audio, verbose=False)
     out = io.StringIO()