nadim71 commited on
Commit
906409a
·
verified ·
1 Parent(s): 92fadab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
  )