Stevenfunau commited on
Commit
517a75b
·
verified ·
1 Parent(s): 91b81a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base"
22
- model=LiteLLMModel(model_id=model_id, api_key=os.getenv('HF_API_KEY'))
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
- model=model,
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",