Stevenfunau commited on
Commit
553189e
·
verified ·
1 Parent(s): be81370

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -19,7 +19,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
19
  #model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
20
  #model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
21
  #model=LiteLLMModel(model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base", api_key=os.getenv('HF_API_KEY'))
22
- model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv('GEMINI_API_KEY'))
 
23
 
24
  # Initialize the model
25
  #model = HfApiModel(model_id)
@@ -33,9 +34,9 @@ class BasicAgent:
33
  self.agent = CodeAgent(
34
 
35
  tools=[
36
- #GoogleSearchTool(provider="serper"),
37
- #VisitWebpageTool()
38
- DuckDuckGoSearchTool()
39
  ],
40
  model=model,
41
  #additional_authorized_imports=[
 
19
  #model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
20
  #model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
21
  #model=LiteLLMModel(model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base", api_key=os.getenv('HF_API_KEY'))
22
+ #model=LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv('GEMINI_API_KEY'))
23
+ model=OpenAIServerModel(model_id="gpt-4o"),
24
 
25
  # Initialize the model
26
  #model = HfApiModel(model_id)
 
34
  self.agent = CodeAgent(
35
 
36
  tools=[
37
+ GoogleSearchTool(provider="serper"),
38
+ VisitWebpageTool()
39
+ #DuckDuckGoSearchTool()
40
  ],
41
  model=model,
42
  #additional_authorized_imports=[