jonnatakusuma commited on
Commit
b872bd7
·
1 Parent(s): a742fdc
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -35,13 +35,15 @@ import ffmpeg
35
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Tokenizer
36
  import librosa
37
 
 
 
38
  st.title("TemplarX-Medium-Indonesian Transcription App")
39
  st.text("Model Whisper (TemplarX-medium-Indonesian) telah dimuat:")
40
 
41
  def load_whisper_model():
42
  model_name = "jonnatakusuma/TemplarX-medium-Indonesian"
43
  tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name)
44
- model = Wav2Vec2ForCTC.from_pretrained(model_name)
45
  return tokenizer, model
46
 
47
  audio_file = st.file_uploader("Unggah Meeting Audio", type=["mp3", "wav", "m4a"])
 
35
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Tokenizer
36
  import librosa
37
 
38
+ HF_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
39
+
40
  st.title("TemplarX-Medium-Indonesian Transcription App")
41
  st.text("Model Whisper (TemplarX-medium-Indonesian) telah dimuat:")
42
 
43
  def load_whisper_model():
44
  model_name = "jonnatakusuma/TemplarX-medium-Indonesian"
45
  tokenizer = Wav2Vec2Tokenizer.from_pretrained(model_name)
46
+ model = Wav2Vec2ForCTC.from_pretrained(model_name, token=HF_TOKEN)
47
  return tokenizer, model
48
 
49
  audio_file = st.file_uploader("Unggah Meeting Audio", type=["mp3", "wav", "m4a"])