lovepreetsingh1996 commited on
Commit
4946de1
·
1 Parent(s): b539f05

fix: try better model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
32
  class BasicAgent:
33
  def __init__(self):
34
  # self.llm = Ollama(model="qwen2.5:7b", request_timeout=500)
35
- self.llm = Gemini(model_name="models/gemini-2.0-flash-lite")
36
  print("BasicAgent initialized.")
37
  def __call__(self, question: str) -> str:
38
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
32
  class BasicAgent:
33
  def __init__(self):
34
  # self.llm = Ollama(model="qwen2.5:7b", request_timeout=500)
35
+ self.llm = Gemini(model_name="models/gemini-2.0-flash")
36
  print("BasicAgent initialized.")
37
  def __call__(self, question: str) -> str:
38
  print(f"Agent received question (first 50 chars): {question[:50]}...")