ak3ra commited on
Commit
5de9f4c
·
1 Parent(s): b95afd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -4,6 +4,7 @@ import torch
4
  import librosa
5
  import json
6
  import os
 
7
 
8
  # with open('ISO_codes.json', 'r') as file:
9
  # iso_codes = json.load(file)
@@ -11,6 +12,9 @@ import os
11
  languages = ["lug", "ach", "nyn", "teo"]
12
  auth_token = os.environ.get("HF_TOKEN")
13
 
 
 
 
14
  model_id = "Sunbird/sunbird-mms"
15
  model = Wav2Vec2ForCTC.from_pretrained(model_id, use_auth_token=auth_token)
16
  processor = Wav2Vec2Processor.from_pretrained(model_id, use_auth_token=auth_token)
 
4
  import librosa
5
  import json
6
  import os
7
+ import huggingface_hub
8
 
9
  # with open('ISO_codes.json', 'r') as file:
10
  # iso_codes = json.load(file)
 
12
  languages = ["lug", "ach", "nyn", "teo"]
13
  auth_token = os.environ.get("HF_TOKEN")
14
 
15
+ from huggingface_hub import login
16
+ login(token=auth_token)
17
+
18
  model_id = "Sunbird/sunbird-mms"
19
  model = Wav2Vec2ForCTC.from_pretrained(model_id, use_auth_token=auth_token)
20
  processor = Wav2Vec2Processor.from_pretrained(model_id, use_auth_token=auth_token)