Update app.py

app.py

@@ -8,7 +8,8 @@ import spaces
 from transformers import pipeline
 from huggingface_hub import model_info
 
-MODEL_NAME = os.environ.get("MODEL_NAME", "NbAiLab/whisper-large-sme")
+MODEL_NAME = os.environ.get("MODEL_NAME", "NbAiLab/salmon-whisper-large-smj-lr7e-5")
+MODEL_VERSION = os.environ.get("MODEL_VERSION", None)
 lang = "fi"
 
 share = (os.environ.get("SHARE", "False")[0].lower() in "ty1") or None
@@ -21,6 +22,7 @@ def pipe(file, return_timestamps=False):
     asr = pipeline(
         task="automatic-speech-recognition",
         model=MODEL_NAME,
+        revision=MODEL_VERSION,
         chunk_length_s=30,
         device=device,
         token=auth_token,