Shivangsinha commited on
Commit
7de61df
·
verified ·
1 Parent(s): 7170c76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -5,7 +5,7 @@ import inspect
5
  import pandas as pd
6
  from smolagents import (
7
  CodeAgent,
8
- InferenceClientModel,
9
  DuckDuckGoSearchTool,
10
  WikipediaSearchTool,
11
  PythonInterpreterTool,
@@ -28,11 +28,11 @@ def get_current_date_time() -> str:
28
  class BasicAgent:
29
  def __init__(self):
30
  print("BasicAgent initialized.")
31
- hf_token = os.getenv("HF_TOKEN")
32
- model = InferenceClientModel(
33
- model_id="meta-llama/Llama-3.1-8B-Instruct",
34
- token=hf_token,
35
- provider="cerebras",
36
  )
37
  tools = [
38
  DuckDuckGoSearchTool(),
 
5
  import pandas as pd
6
  from smolagents import (
7
  CodeAgent,
8
+ OpenAIServerModel,
9
  DuckDuckGoSearchTool,
10
  WikipediaSearchTool,
11
  PythonInterpreterTool,
 
28
  class BasicAgent:
29
  def __init__(self):
30
  print("BasicAgent initialized.")
31
+ sambanova_api_key = os.getenv("SAMBANOVA_API_KEY")
32
+ model = OpenAIServerModel(
33
+ model_id="Meta-Llama-3.3-70B-Instruct",
34
+ api_base="https://api.sambanova.ai/v1",
35
+ api_key=sambanova_api_key,
36
  )
37
  tools = [
38
  DuckDuckGoSearchTool(),