Update app.py
Browse files
app.py
CHANGED
|
@@ -1,12 +1,8 @@
|
|
| 1 |
-
from smolagents import CodeAgent,
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|