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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -14,7 +14,7 @@ from smolagents import Tool, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchToo
14
  # --- Constants ---
15
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
16
  model_qwen = HfApiModel(
17
- model_id='Qwen/Qwen-VL-Chat',
18
  max_tokens=2048, # Reasonable default for this model
19
  temperature=0.2, # You can adjust this based on how creative you want answers to be
20
  custom_role_conversions=None
@@ -100,9 +100,7 @@ class VisitWebpageTool(Tool):
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']
 
14
  # --- Constants ---
15
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
16
  model_qwen = HfApiModel(
17
+ model_id='llava-hf/llava-1.5-13b-hf',
18
  max_tokens=2048, # Reasonable default for this model
19
  temperature=0.2, # You can adjust this based on how creative you want answers to be
20
  custom_role_conversions=None
 
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']