Update model.py
Browse files
model.py
CHANGED
|
@@ -11,7 +11,7 @@ your_endpoint_url = "https://kp4xdy196cw81uf3.us-east-1.aws.endpoints.huggingfac
|
|
| 11 |
|
| 12 |
llm = HuggingFaceEndpoint(
|
| 13 |
endpoint_url=f"{your_endpoint_url}",
|
| 14 |
-
huggingfacehub_api_token = ,
|
| 15 |
task = "text-generation",
|
| 16 |
max_new_tokens=128,
|
| 17 |
top_k=10,
|
|
|
|
| 11 |
|
| 12 |
llm = HuggingFaceEndpoint(
|
| 13 |
endpoint_url=f"{your_endpoint_url}",
|
| 14 |
+
huggingfacehub_api_token = st.secrets["HUGGINGFACEHUB_API_TOKEN"],
|
| 15 |
task = "text-generation",
|
| 16 |
max_new_tokens=128,
|
| 17 |
top_k=10,
|