mojad121 commited on
Commit
369fd9f
·
verified ·
1 Parent(s): 4ef6d8e

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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["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)
 
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)