Alimjoo commited on
Commit
8c4b7c0
·
1 Parent(s): f1f846b

added hf_token

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -9,6 +9,9 @@ from pydantic import BaseModel
9
  import string
10
  import unicodedata
11
 
 
 
 
12
 
13
  app = FastAPI()
14
 
@@ -89,8 +92,8 @@ def fix_string(batch):
89
 
90
  # model = VitsModel.from_pretrained("facebook/mms-tts-uig-script_arabic")
91
  # tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-uig-script_arabic")
92
- model_ug = VitsModel.from_pretrained("piyazon/TTS-Radio")
93
- tokenizer_ug = AutoTokenizer.from_pretrained("piyazon/TTS-Radio")
94
  # model_ug = VitsModel.from_pretrained("piyazon/qutadgu_bilik")
95
  # tokenizer_ug = AutoTokenizer.from_pretrained("piyazon/qutadgu_bilik")
96
 
 
9
  import string
10
  import unicodedata
11
 
12
+ import os
13
+ # Access the secret named "MY_API_KEY"
14
+ hf_token = os.environ.get("HF_TOKEN")
15
 
16
  app = FastAPI()
17
 
 
92
 
93
  # model = VitsModel.from_pretrained("facebook/mms-tts-uig-script_arabic")
94
  # tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-uig-script_arabic")
95
+ model_ug = VitsModel.from_pretrained("piyazon/TTS-Radio", token=hf_token)
96
+ tokenizer_ug = AutoTokenizer.from_pretrained("piyazon/TTS-Radio", token=hf_token)
97
  # model_ug = VitsModel.from_pretrained("piyazon/qutadgu_bilik")
98
  # tokenizer_ug = AutoTokenizer.from_pretrained("piyazon/qutadgu_bilik")
99