ragunath-ravi commited on
Commit
b6265ab
·
verified ·
1 Parent(s): d8ff3ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -5,6 +5,13 @@ from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
5
  import numpy as np
6
  import librosa
7
 
 
 
 
 
 
 
 
8
  # Load model
9
  processor = AutoProcessor.from_pretrained("ragunath-ravi/whisper-mini-ta")
10
  model = AutoModelForSpeechSeq2Seq.from_pretrained("ragunath-ravi/whisper-mini-ta")
 
5
  import numpy as np
6
  import librosa
7
 
8
+ import os
9
+ from huggingface_hub import login
10
+
11
+ hf_token = os.getenv("Token")
12
+ if hf_token is None:
13
+ raise ValueError("HF_TOKEN environment variable not set.")
14
+ login(token=hf_token)
15
  # Load model
16
  processor = AutoProcessor.from_pretrained("ragunath-ravi/whisper-mini-ta")
17
  model = AutoModelForSpeechSeq2Seq.from_pretrained("ragunath-ravi/whisper-mini-ta")