MADtoBAD commited on
Commit
3cf5415
·
verified ·
1 Parent(s): 40f9320

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -1,12 +1,8 @@
1
- from smolagents import CodeAgent, HfApiModel, load_tool, tool, DuckDuckGoSearchTool
2
  import datetime
3
  import requests
4
  import pytz
5
  import yaml
6
- import base64
7
- import os
8
- import io
9
- from PIL import Image
10
  from tools.final_answer import FinalAnswerTool
11
  from Gradio_UI import GradioUI
12
 
@@ -123,11 +119,14 @@ def suggest_weather_sources(city: str) -> str:
123
 
124
  final_answer = FinalAnswerTool()
125
 
126
- model = HfApiModel(
 
 
 
 
 
127
  max_tokens=2096,
128
- temperature=0.5,
129
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
130
- custom_role_conversions=None,
131
  )
132
 
133
  with open("prompts.yaml", 'r') as stream:
 
1
+ from smolagents import CodeAgent, DuckDuckGoSearchTool
2
  import datetime
3
  import requests
4
  import pytz
5
  import yaml
 
 
 
 
6
  from tools.final_answer import FinalAnswerTool
7
  from Gradio_UI import GradioUI
8
 
 
119
 
120
  final_answer = FinalAnswerTool()
121
 
122
+ # Используем локальную модель вместо HfApiModel
123
+ from transformers import AutoModelForCausalLM, AutoTokenizer
124
+ from smolagents import TransformersModel
125
+
126
+ model = TransformersModel(
127
+ model_id='Qwen/Qwen2.5-Coder-1.5B', # Используем меньшую модель для совместимости
128
  max_tokens=2096,
129
+ temperature=0.5
 
 
130
  )
131
 
132
  with open("prompts.yaml", 'r') as stream: