Maitreyee22 commited on
Commit
23ed344
·
verified ·
1 Parent(s): 915b070

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -14,8 +14,8 @@ import streamlit as st
14
  # It simplifies loading models/tokenizers and running common tasks
15
  from transformers import pipeline
16
 
17
- hf_token = st.secrets["HF_HUB_TOKEN"]
18
- #api_key = os.getenv("HF_HUB_TOKEN")
19
 
20
  # 1. Cache the pipeline so it loads once
21
  @st.cache_resource
 
14
  # It simplifies loading models/tokenizers and running common tasks
15
  from transformers import pipeline
16
 
17
+ #hf_token = st.secrets["HF_HUB_TOKEN"]
18
+ hf_token= os.getenv("HUGGING_FACE_HUB_TOKEN")
19
 
20
  # 1. Cache the pipeline so it loads once
21
  @st.cache_resource