Spaces:
Runtime error
Runtime error
Commit
·
e834ef6
1
Parent(s):
3459d0e
Update main.py
Browse files
main.py
CHANGED
|
@@ -26,13 +26,13 @@ import requests
|
|
| 26 |
import uuid
|
| 27 |
import sys
|
| 28 |
|
| 29 |
-
|
| 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=
|
| 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,
|