Model update
#13
by
sdeepanraj
- opened
app.py
CHANGED
@@ -5,11 +5,16 @@ import inspect
|
|
5 |
import pandas as pd
|
6 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, InferenceClientModel, WikipediaSearchTool, VisitWebpageTool
|
7 |
from huggingface_hub import login
|
|
|
|
|
8 |
|
9 |
# (Keep Constants as is)
|
10 |
# --- Constants ---
|
11 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
12 |
|
|
|
|
|
|
|
13 |
# --- Basic Agent Definition ---
|
14 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
15 |
class BasicAgent:
|
@@ -23,7 +28,7 @@ class BasicAgent:
|
|
23 |
# custom_role_conversions=None,
|
24 |
#)
|
25 |
|
26 |
-
model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together", max_tokens=2096)
|
27 |
self.agent = CodeAgent(
|
28 |
model = model,
|
29 |
tools = [DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool()],
|
|
|
5 |
import pandas as pd
|
6 |
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, InferenceClientModel, WikipediaSearchTool, VisitWebpageTool
|
7 |
from huggingface_hub import login
|
8 |
+
import os
|
9 |
+
|
10 |
|
11 |
# (Keep Constants as is)
|
12 |
# --- Constants ---
|
13 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
14 |
|
15 |
+
token = os.getenv('HF_API_TOKEN')
|
16 |
+
|
17 |
+
|
18 |
# --- Basic Agent Definition ---
|
19 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
20 |
class BasicAgent:
|
|
|
28 |
# custom_role_conversions=None,
|
29 |
#)
|
30 |
|
31 |
+
model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together", max_tokens=2096, token = token)
|
32 |
self.agent = CodeAgent(
|
33 |
model = model,
|
34 |
tools = [DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool()],
|