Aakash010 commited on
Commit
be81334
·
verified ·
1 Parent(s): 195589b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -18,9 +18,10 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
19
  class BasicAgent:
20
  def __init__(self):
21
- model =model = LiteLLMModel(
22
- model_id="gemini/gemini-2.0-flash-lite",
23
- api_key=os.getenv("GEMINI_API_KEY")
 
24
  )
25
  self.agent = CodeAgent(
26
  model=model,
 
18
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
19
  class BasicAgent:
20
  def __init__(self):
21
+ model =OpenAIServerModel(
22
+ model_id="llama-3.3-70b-versatile",
23
+ api_base="https://api.groq.com/openai/v1",
24
+ api_key=os.getenv("GROQ_API_KEY")
25
  )
26
  self.agent = CodeAgent(
27
  model=model,