snnorty commited on
Commit
f253e00
·
verified ·
1 Parent(s): 4542327

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -2,15 +2,16 @@ import os
2
  import requests
3
  import pandas as pd
4
  import gradio as gr
5
- from smolagents import CodeAgent, DuckDuckGoSearchTool, LiteLLMModel
6
 
7
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
8
- MODEL_ID = "huggingface/Qwen/Qwen2.5-Coder-32B-Instruct"
9
 
10
  def create_agent(token: str):
11
- model = LiteLLMModel(
 
12
  model_id=MODEL_ID,
13
- api_key=token
14
  )
15
  search_tool = DuckDuckGoSearchTool()
16
 
@@ -19,9 +20,9 @@ def create_agent(token: str):
19
  model=model,
20
  additional_authorized_imports=[
21
  "requests", "bs4", "re", "math", "datetime",
22
- "json", "pandas", "numpy", "urllib", "bs4"
23
  ],
24
- max_steps=12, # Увеличили с 6 до 12, чтобы агент успевал завершать сложные задачи
25
  verbosity_level=1
26
  )
27
  return agent
 
2
  import requests
3
  import pandas as pd
4
  import gradio as gr
5
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
6
 
7
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
8
+ MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
9
 
10
  def create_agent(token: str):
11
+ # Используем HfApiModel вместо LiteLLMModel, чтобы избежать ошибок с установкой дополнительных библиотек
12
+ model = HfApiModel(
13
  model_id=MODEL_ID,
14
+ token=token
15
  )
16
  search_tool = DuckDuckGoSearchTool()
17
 
 
20
  model=model,
21
  additional_authorized_imports=[
22
  "requests", "bs4", "re", "math", "datetime",
23
+ "json", "pandas", "numpy", "urllib"
24
  ],
25
+ max_steps=12, # Увеличенный лимит шагов для сложных задач
26
  verbosity_level=1
27
  )
28
  return agent