ManasSharma07 commited on
Commit
d3d8c78
·
verified ·
1 Parent(s): 57844b6

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +9 -4
src/streamlit_app.py CHANGED
@@ -16,12 +16,17 @@ import pandas as pd
16
 
17
  from huggingface_hub import login
18
 
 
 
 
 
 
 
19
  try:
20
- hf_token = st.secrets["HF_TOKEN"]["token"]
21
- os.environ["HF_TOKEN"] = hf_token
22
- login(token=hf_token)
23
  except Exception as e:
24
- print("streamlit hf secret not defined/assigned")
25
 
26
  import os
27
  os.environ["STREAMLIT_WATCHER_TYPE"] = "none"
 
16
 
17
  from huggingface_hub import login
18
 
19
+ # try:
20
+ # hf_token = st.secrets["HF_TOKEN"]["token"]
21
+ # os.environ["HF_TOKEN"] = hf_token
22
+ # login(token=hf_token)
23
+ # except Exception as e:
24
+ # print("streamlit hf secret not defined/assigned")
25
  try:
26
+ hf_token = os.getenv("YOUR SECRET KEY")
27
+ login(token = hf_token)
 
28
  except Exception as e:
29
+ print("hf secret not defined/assigned")
30
 
31
  import os
32
  os.environ["STREAMLIT_WATCHER_TYPE"] = "none"