MissingBreath commited on
Commit
e8ac00d
·
verified ·
1 Parent(s): efc9338

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +4 -0
api.py CHANGED
@@ -5,6 +5,10 @@ import io
5
  import tensorflow as tf
6
  import os
7
  # os.environ['HF_TOKEN']=''
 
 
 
 
8
 
9
 
10
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
5
  import tensorflow as tf
6
  import os
7
  # os.environ['HF_TOKEN']=''
8
+ from huggingface_hub import login
9
+
10
+ hf_token = os.getenv("HF_TOKEN")
11
+ login(token=hf_token)
12
 
13
 
14
  from transformers import AutoTokenizer, AutoModelForSequenceClassification