lhoestq HF staff commited on
Commit
65c21a7
·
1 Parent(s): 27e089d

add HF_TOKEN_INFERENCE_API

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import json
2
  import time
3
  from itertools import count, islice
@@ -17,7 +18,7 @@ from utils import StringIteratorIO
17
 
18
 
19
  model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
20
- client = InferenceClient(model_id)
21
 
22
  session = requests.Session()
23
  empty_dataframe = pd.DataFrame({"1": [], "2": [], "3": []})
 
1
+ import os
2
  import json
3
  import time
4
  from itertools import count, islice
 
18
 
19
 
20
  model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
21
+ client = InferenceClient(model_id, token=os.environ.get("HF_TOKEN_INFERENCE_API"))
22
 
23
  session = requests.Session()
24
  empty_dataframe = pd.DataFrame({"1": [], "2": [], "3": []})