choco-conoz commited on
Commit
a6587b7
·
1 Parent(s): 216b57a

feat: add env

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +4 -0
src/streamlit_app.py CHANGED
@@ -1,8 +1,12 @@
 
1
  import streamlit as st
2
  import torch
3
  from transformers import pipeline
4
  # from unsloth import FastLanguageModel, is_bfloat16_supported
5
 
 
 
 
6
 
7
  def main():
8
  st.title('Text Generator')
 
1
+ import os
2
  import streamlit as st
3
  import torch
4
  from transformers import pipeline
5
  # from unsloth import FastLanguageModel, is_bfloat16_supported
6
 
7
+ hf_token = st.secrets["HUGGINGFACE_TOKEN"]["token"]
8
+ os.environ["HUGGINGFACEHUB_API_TOKEN"] = hf_token
9
+
10
 
11
  def main():
12
  st.title('Text Generator')