MrArray22 commited on
Commit
637c891
·
verified ·
1 Parent(s): 747589e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import os
2
  import gradio as gr
3
  import requests
4
  import pandas as pd
5
- from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
6
  # (Keep Constants as is)
7
  # --- Constants ---
8
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -15,7 +15,8 @@ class BasicAgent:
15
  print("BasicAgent initialized.")
16
  # Initialize the model
17
  #model = HfApiModel()
18
- model = OpenAIServerModel(model_id="openai/gpt-4o",api_key=os.environ["API_KEY"],api_base="https://models.github.ai/inference")
 
19
  # Initialize the search tool
20
  search_tool = DuckDuckGoSearchTool()
21
  # Initialize Agent
 
2
  import gradio as gr
3
  import requests
4
  import pandas as pd
5
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel,HfApiModel
6
  # (Keep Constants as is)
7
  # --- Constants ---
8
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
15
  print("BasicAgent initialized.")
16
  # Initialize the model
17
  #model = HfApiModel()
18
+ model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct",api_key=os.environ['HF_TOKEN'])
19
+ # model = OpenAIServerModel(model_id="openai/gpt-4o",api_key=os.environ["API_KEY"],api_base="https://models.github.ai/inference")
20
  # Initialize the search tool
21
  search_tool = DuckDuckGoSearchTool()
22
  # Initialize Agent