ArunKr commited on
Commit
d27e3b8
·
1 Parent(s): 935b2ae

Model updated

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -22,12 +22,18 @@ class GAIAAgent:
22
  # api_key=api_key
23
  # )
24
 
25
-
26
- self.model = OpenAIServerModel(
 
27
  model_id="Qwen/Qwen3-235B-A22B",
28
- api_base="https://api.studio.nebius.com/", # Leave this blank to query OpenAI servers.
29
- api_key=api_key, # Switch to the API key for the server you're targeting.
30
- )
 
 
 
 
 
31
  self.tools = [DuckDuckGoSearchTool(), VisitWebpageTool()]
32
  self.agent = CodeAgent(tools=self.tools, model=self.model, max_steps=10)
33
 
 
22
  # api_key=api_key
23
  # )
24
 
25
+ # from smolagents import InferenceClientModel
26
+
27
+ self.model = InferenceClientModel(
28
  model_id="Qwen/Qwen3-235B-A22B",
29
+ provider="nebius")
30
+
31
+ # self.model = OpenAIServerModel(
32
+ # model_id="Qwen/Qwen3-235B-A22B",
33
+ # api_base="https://api.studio.nebius.com/", # Leave this blank to query OpenAI servers.
34
+ # api_key=api_key, # Switch to the API key for the server you're targeting.
35
+ # )
36
+
37
  self.tools = [DuckDuckGoSearchTool(), VisitWebpageTool()]
38
  self.agent = CodeAgent(tools=self.tools, model=self.model, max_steps=10)
39