Antoine101 commited on
Commit
a5c3011
·
verified ·
1 Parent(s): e217ebb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -9,6 +9,7 @@ from langgraph.graph import START, StateGraph, MessagesState
9
  from langgraph.graph.message import add_messages
10
  from langgraph.prebuilt import ToolNode, tools_condition
11
  from langchain_core.messages import AnyMessage, HumanMessage
 
12
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
13
 
14
  # (Keep Constants as is)
@@ -16,7 +17,7 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
16
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
17
 
18
  # --- Basic Agent Definition ---
19
-
20
  #llm = HuggingFaceEndpoint(repo_id="HuggingFaceTB/SmolLM-135M-Instruct")
21
  llm = HuggingFaceEndpoint(
22
  repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
@@ -25,6 +26,7 @@ llm = HuggingFaceEndpoint(
25
  do_sample=False,
26
  repetition_penalty=1.03,
27
  temperature=0,
 
28
  )
29
 
30
  chat = ChatHuggingFace(llm=llm, verbose=True)
 
9
  from langgraph.graph.message import add_messages
10
  from langgraph.prebuilt import ToolNode, tools_condition
11
  from langchain_core.messages import AnyMessage, HumanMessage
12
+ from langchain_groq import ChatGroq
13
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
14
 
15
  # (Keep Constants as is)
 
17
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
 
19
  # --- Basic Agent Definition ---
20
+ #llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
21
  #llm = HuggingFaceEndpoint(repo_id="HuggingFaceTB/SmolLM-135M-Instruct")
22
  llm = HuggingFaceEndpoint(
23
  repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
 
26
  do_sample=False,
27
  repetition_penalty=1.03,
28
  temperature=0,
29
+ huggingfacehub_api_token=HF_TOKEN
30
  )
31
 
32
  chat = ChatHuggingFace(llm=llm, verbose=True)