Update app.py

app.py

@@ -8,8 +8,8 @@ import spaces
 from transformers import pipeline
 from huggingface_hub import model_info
 
-MODEL_NAME = "NbAiLab/whisper-large-sme" #this always needs to stay in line 8 :D sorry for the hackiness
-lang = "fi"
+MODEL_NAME = "NbAiLab/nb-whisper-large"
+lang = "no"
 
 share = (os.environ.get("SHARE", "False")[0].lower() in "ty1") or None
 auth_token = os.environ.get("AUTH_TOKEN") or True
@@ -30,7 +30,7 @@ def pipe(file, return_timestamps=False):
         task="transcribe",
         no_timestamps=not return_timestamps,
     )
-    # asr.model.config.no_timestamps_token_id = asr.tokenizer.encode("<|notimestamps|>", add_special_tokens=False)[0]
+    asr.model.config.no_timestamps_token_id = asr.tokenizer.encode("<|notimestamps|>", add_special_tokens=False)[0]
     return asr(file, return_timestamps=return_timestamps)
 
 def transcribe(file, return_timestamps=False):
@@ -74,7 +74,7 @@ mf_transcribe = gr.Interface(
     fn=transcribe,
     inputs=[
         gr.components.Audio(sources=['upload', 'microphone'], type="filepath"),
-        # gr.components.Checkbox(label="Return timestamps"),
+        gr.components.Checkbox(label="Return timestamps"),
     ],
     outputs="text",
     theme="huggingface",
@@ -91,7 +91,7 @@ yt_transcribe = gr.Interface(
     fn=yt_transcribe,
     inputs=[
         gr.components.Textbox(lines=1, placeholder="Paste the URL to a YouTube video here", label="YouTube URL"),
-        # gr.components.Checkbox(label="Return timestamps"),
+        gr.components.Checkbox(label="Return timestamps"),
     ],
     examples=[["https://www.youtube.com/watch?v=mukeSSa5GKo"]],
     outputs=["html", "text"],