Trying InferenceClientModel (#11)
Browse files- Trying InferenceClientModel (9f2e12a37a5462839347e8b0a52fce7fa669d8b8)
app.py
CHANGED
@@ -3,7 +3,8 @@ import gradio as gr
|
|
3 |
import requests
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
-
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, WikipediaSearchTool, VisitWebpageTool
|
|
|
7 |
|
8 |
# (Keep Constants as is)
|
9 |
# --- Constants ---
|
@@ -14,12 +15,15 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
14 |
class BasicAgent:
|
15 |
def __init__(self):
|
16 |
print("BasicAgent initialized.")
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
|
|
|
|
|
|
23 |
self.agent = CodeAgent(
|
24 |
model = model,
|
25 |
tools = [DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool()],
|
|
|
3 |
import requests
|
4 |
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 ---
|
|
|
15 |
class BasicAgent:
|
16 |
def __init__(self):
|
17 |
print("BasicAgent initialized.")
|
18 |
+
login()
|
19 |
+
#model = HfApiModel(
|
20 |
+
# max_tokens=2096,
|
21 |
+
# temperature=0.5,
|
22 |
+
# model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
|
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()],
|