Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
@@ -8,9 +8,9 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
8 |
os.environ["TRANSFORMERS_CACHE"] = "/app/.cache/huggingface"
|
9 |
os.environ["HF_HOME"] = "/app/.cache/huggingface"
|
10 |
os.environ["TORCH_HOME"] = "/app/.cache/torch"
|
11 |
-
os.environ["HF_TOKEN"] = st.secrets["
|
12 |
|
13 |
-
hf_token = st.secrets["
|
14 |
|
15 |
whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny", token=hf_token)
|
16 |
whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny", token=hf_token)
|
|
|
8 |
os.environ["TRANSFORMERS_CACHE"] = "/app/.cache/huggingface"
|
9 |
os.environ["HF_HOME"] = "/app/.cache/huggingface"
|
10 |
os.environ["TORCH_HOME"] = "/app/.cache/torch"
|
11 |
+
os.environ["HF_TOKEN"] = st.secrets["Hf_token"]
|
12 |
|
13 |
+
hf_token = st.secrets["Hf_token"]
|
14 |
|
15 |
whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny", token=hf_token)
|
16 |
whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny", token=hf_token)
|