Files changed (1) hide show
  1. app.py +6 -1
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()],