Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -2
src/streamlit_app.py
CHANGED
@@ -8,8 +8,7 @@ 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 |
-
|
12 |
-
hf_token = st.secrets["HateSpeechMujtabatoken"]
|
13 |
|
14 |
whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny", token=hf_token)
|
15 |
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 |
+
hf_token = os.getenv("HateSpeechMujtabatoken")
|
|
|
12 |
|
13 |
whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny", token=hf_token)
|
14 |
whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny", token=hf_token)
|