Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,6 +11,8 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace, HuggingF
|
|
| 11 |
from langchain_core.messages import HumanMessage
|
| 12 |
|
| 13 |
|
|
|
|
|
|
|
| 14 |
|
| 15 |
# (Keep Constants as is)
|
| 16 |
# --- Constants ---
|
|
@@ -32,7 +34,6 @@ class MyAgent:
|
|
| 32 |
self.llm = ChatHuggingFace(
|
| 33 |
llm=HuggingFaceEndpoint(
|
| 34 |
repo_id="mistralai/Mistral-7B-Instruct-v0.2",
|
| 35 |
-
huggingfacehub_api_token='HF_API',
|
| 36 |
task="text-generation",
|
| 37 |
provider="hf-inference",
|
| 38 |
)
|
|
|
|
| 11 |
from langchain_core.messages import HumanMessage
|
| 12 |
|
| 13 |
|
| 14 |
+
HF_API=os.getenv('HUGGINGFACEHUB_API_TOKEN'))
|
| 15 |
+
|
| 16 |
|
| 17 |
# (Keep Constants as is)
|
| 18 |
# --- Constants ---
|
|
|
|
| 34 |
self.llm = ChatHuggingFace(
|
| 35 |
llm=HuggingFaceEndpoint(
|
| 36 |
repo_id="mistralai/Mistral-7B-Instruct-v0.2",
|
|
|
|
| 37 |
task="text-generation",
|
| 38 |
provider="hf-inference",
|
| 39 |
)
|