binqiangliu commited on
Commit
e834ef6
·
1 Parent(s): 3459d0e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -26,13 +26,13 @@ import requests
26
  import uuid
27
  import sys
28
 
29
- HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
30
  repo_id = os.environ.get('repo_id')
31
  #port = os.getenv('port')
32
 
33
  llm = HuggingFaceHub(repo_id=repo_id,
34
  #huggingfacehub_api_token="hf_p***K",
35
- huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
36
  model_kwargs={"min_length":1024,
37
  "max_new_tokens":5632, "do_sample":True,
38
  "temperature":0.1,
 
26
  import uuid
27
  import sys
28
 
29
+ hf_token = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
30
  repo_id = os.environ.get('repo_id')
31
  #port = os.getenv('port')
32
 
33
  llm = HuggingFaceHub(repo_id=repo_id,
34
  #huggingfacehub_api_token="hf_p***K",
35
+ huggingfacehub_api_token=hf_token,
36
  model_kwargs={"min_length":1024,
37
  "max_new_tokens":5632, "do_sample":True,
38
  "temperature":0.1,