sdeepanraj commited on
Commit
7ed01bf
·
verified ·
1 Parent(s): 334fdc8

Trying InferenceClientModel (#11)

Browse files

- Trying InferenceClientModel (9f2e12a37a5462839347e8b0a52fce7fa669d8b8)

Files changed (1) hide show
  1. app.py +11 -7
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
- model = HfApiModel(
18
- max_tokens=2096,
19
- temperature=0.5,
20
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
21
- custom_role_conversions=None,
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()],