wt002 commited on
Commit
df62ccf
·
verified ·
1 Parent(s): 9363094

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -4,13 +4,8 @@ import time
4
  import gradio as gr
5
  import pandas as pd
6
  import requests
7
- from smolagents import (
8
- CodeAgent,
9
- DuckDuckGoSearchTool,
10
- LiteLLMModel,
11
- WikipediaSearchTool,
12
- FinalAnswerTool
13
- )
14
 
15
  from agent import (
16
  analyze_audio_file,
@@ -31,7 +26,7 @@ class BasicAgent:
31
 
32
  # Initialize the model
33
  model = LiteLLMModel(model_id=os.getenv("MODEL_ID"),
34
- api_key=os.getenv("API_KEY"))
35
 
36
  # Initialize the searchs tool
37
  duck_duck_go_search_tool = DuckDuckGoSearchTool()
 
4
  import gradio as gr
5
  import pandas as pd
6
  import requests
7
+ from smolagents import tool, Tool, CodeAgent, DuckDuckGoSearchTool, LiteLLMModel, HfApiModel, VisitWebpageTool, SpeechToTextTool, FinalAnswerTool
8
+
 
 
 
 
 
9
 
10
  from agent import (
11
  analyze_audio_file,
 
26
 
27
  # Initialize the model
28
  model = LiteLLMModel(model_id=os.getenv("MODEL_ID"),
29
+ api_key=os.getenv("GOOGLE_API_KEY"))
30
 
31
  # Initialize the searchs tool
32
  duck_duck_go_search_tool = DuckDuckGoSearchTool()