antonioschiro commited on
Commit
506a878
·
verified ·
1 Parent(s): 766e143

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -42,12 +42,15 @@ search_tool = DuckDuckGoSearchRun()
42
  tools = [search_tool]
43
 
44
  # LLM
45
- model = ChatHuggingFace(
46
- llm=HuggingFaceEndpoint(
47
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
48
- temperature=0,
49
- ),
 
50
  )
 
 
51
  class BasicAgent:
52
  def __init__(self):
53
  self.agent = create_react_agent(model, tools, prompt)
 
42
  tools = [search_tool]
43
 
44
  # LLM
45
+ llm = HuggingFaceEndpoint(
46
+ repo_id="deepseek-ai/DeepSeek-R1-0528",
47
+ task="text-generation",
48
+ provider="hyperbolic", # set your provider here
49
+ # provider="nebius",
50
+ # provider="together",
51
  )
52
+ model = ChatHuggingFace(llm=llm)
53
+
54
  class BasicAgent:
55
  def __init__(self):
56
  self.agent = create_react_agent(model, tools, prompt)