Update app.py
Browse files
app.py
CHANGED
@@ -18,8 +18,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
18 |
#--SM
|
19 |
#model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
|
20 |
#model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
|
21 |
-
|
22 |
-
model=LiteLLMModel(
|
23 |
|
24 |
# Initialize the model
|
25 |
#model = HfApiModel(model_id)
|
@@ -31,11 +31,12 @@ model=LiteLLMModel(model_id=model_id, api_key=os.getenv('HF_API_KEY'))
|
|
31 |
class BasicAgent:
|
32 |
def __init__(self):
|
33 |
self.agent = CodeAgent(
|
34 |
-
|
35 |
tools=[
|
36 |
GoogleSearchTool(provider="serper"),
|
37 |
VisitWebpageTool()
|
38 |
],
|
|
|
39 |
#additional_authorized_imports=[
|
40 |
# "geopandas",
|
41 |
# "plotly",
|
|
|
18 |
#--SM
|
19 |
#model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
|
20 |
#model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
|
21 |
+
|
22 |
+
model=LiteLLMModel(model="DeepSeek-Coder-V2-Lite-Base", provider="huggingface", api_key=os.getenv('HF_API_KEY'))
|
23 |
|
24 |
# Initialize the model
|
25 |
#model = HfApiModel(model_id)
|
|
|
31 |
class BasicAgent:
|
32 |
def __init__(self):
|
33 |
self.agent = CodeAgent(
|
34 |
+
|
35 |
tools=[
|
36 |
GoogleSearchTool(provider="serper"),
|
37 |
VisitWebpageTool()
|
38 |
],
|
39 |
+
model=model,
|
40 |
#additional_authorized_imports=[
|
41 |
# "geopandas",
|
42 |
# "plotly",
|