wt002 commited on
Commit
6c628db
·
verified ·
1 Parent(s): f056616

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
19
 
20
  # --- Basic Agent Definition ---
21
  class BasicAgent:
22
- def __init__(self, model="gemini-2.0-flash-lite"):
23
  self.api_url = f"https://api-inference.huggingface.co/models/{model}"
24
  self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
25
  # Wikipedia setup (with proper User-Agent)
 
19
 
20
  # --- Basic Agent Definition ---
21
  class BasicAgent:
22
+ def __init__(self, model="google/gemma-7b"):
23
  self.api_url = f"https://api-inference.huggingface.co/models/{model}"
24
  self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
25
  # Wikipedia setup (with proper User-Agent)