MariaFilipkowska commited on
Commit
462ab67
·
verified ·
1 Parent(s): 9f2cedc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -13,13 +13,14 @@ from langchain.tools import Tool
13
  from langgraph.graph import START, StateGraph
14
  from langgraph.graph.message import add_messages
15
  from langgraph.prebuilt import ToolNode, tools_condition
16
- from huggingface_hub import InferenceClient
17
- from langchain.chat_models import init_chat_model
 
18
 
19
  print("Loading LLM...")
20
- client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
21
- chat = init_chat_model("llama-3.2-3b-instruct", model_provider="huggingface")
22
- print(f"Model {chat.model} downloaded!")
23
 
24
  # (Keep Constants as is)
25
  # --- Constants ---
 
13
  from langgraph.graph import START, StateGraph
14
  from langgraph.graph.message import add_messages
15
  from langgraph.prebuilt import ToolNode, tools_condition
16
+ HuggingFaceEndpoint,
17
+ from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
18
+
19
 
20
  print("Loading LLM...")
21
+ llm = HuggingFaceEndpoint("meta-llama/Llama-3.2-3B-Instruct")
22
+ chat = ChatHuggingFace("llama-3.2-3b-instruct")
23
+ print(f"Model {chat.llm} downloaded!")
24
 
25
  # (Keep Constants as is)
26
  # --- Constants ---