Update app.py
Browse files
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 =
|
|
|
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
|