Firemedic15 commited on
Commit
402966d
·
verified ·
1 Parent(s): 7e68f51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -29,10 +29,10 @@ from brief import (
29
  MODEL_ID = "Qwen/Qwen2.5-72B-Instruct" # Strong free model on HF Inference
30
 
31
  def build_agent() -> ToolCallingAgent:
32
- model = HfApiModel(
33
- model_id=MODEL_ID,
34
- token=os.environ.get("HF_TOKEN"),
35
- )
36
  agent = ToolCallingAgent(
37
  tools=[fetch_acled_events, fetch_rss_headlines, list_available_sources],
38
  model=model,
 
29
  MODEL_ID = "Qwen/Qwen2.5-72B-Instruct" # Strong free model on HF Inference
30
 
31
  def build_agent() -> ToolCallingAgent:
32
+ model = InferenceClientModel(
33
+ model_id=MODEL_ID,
34
+ token=os.environ.get("HF_TOKEN"),
35
+ )
36
  agent = ToolCallingAgent(
37
  tools=[fetch_acled_events, fetch_rss_headlines, list_available_sources],
38
  model=model,