Ap98 commited on
Commit
32a89a3
·
verified ·
1 Parent(s): ecc3d3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -100,7 +100,9 @@ class VisitWebpageTool(Tool):
100
  class BasicAgent:
101
  def __init__(self):
102
  self.agent = CodeAgent(
103
- model= model_qwen,
 
 
104
  tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool(), DownloadTaskAttachmentTool()],
105
  add_base_tools=True,
106
  additional_authorized_imports=['pandas','numpy','csv','subprocess', 'exec']
 
100
  class BasicAgent:
101
  def __init__(self):
102
  self.agent = CodeAgent(
103
+ model= LiteLLMModel(
104
+ model_id="llava-hf/llava-1.5-13b-hf" # Hugging Face multimodal model
105
+ ),
106
  tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool(), DownloadTaskAttachmentTool()],
107
  add_base_tools=True,
108
  additional_authorized_imports=['pandas','numpy','csv','subprocess', 'exec']