Nienke Dekkema commited on
Commit
39fd908
·
1 Parent(s): ced5ba2
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from smolagents import CodeAgent, UserInputTool, WebSearchTool, InferenceClientModel, PythonInterpreterTool, PythonExecutor, FinalAnswerTool, Tool, tool, LiteLLMModel, DuckDuckGoSearchTool
7
  from dotenv import load_dotenv
8
 
9
  # (Keep Constants as is)
@@ -47,8 +47,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
47
  # PythonExecutor(),
48
  UserInputTool(),
49
  ],
50
- model=LiteLLMModel(
51
- model_id='gpt2'),
 
52
  max_steps=10,
53
  verbosity_level=2
54
  )
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import InferenceClientModel, CodeAgent, UserInputTool, WebSearchTool, InferenceClientModel, PythonInterpreterTool, PythonExecutor, FinalAnswerTool, Tool, tool, LiteLLMModel, DuckDuckGoSearchTool
7
  from dotenv import load_dotenv
8
 
9
  # (Keep Constants as is)
 
47
  # PythonExecutor(),
48
  UserInputTool(),
49
  ],
50
+ # model=LiteLLMModel(
51
+ # model_id='gpt2'),
52
+ model=InferenceClientModel(),
53
  max_steps=10,
54
  verbosity_level=2
55
  )