asthara commited on
Commit
68495e7
·
1 Parent(s): 9f68418

try gemini model

Browse files
Files changed (2) hide show
  1. agent.py +3 -3
  2. app.py +2 -1
agent.py CHANGED
@@ -7,9 +7,9 @@ from smolagents import (
7
  from prompts import prompt_templates
8
 
9
  model = InferenceClientModel(
10
- max_tokens=2096,
11
- temperature=0.5,
12
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
13
  custom_role_conversions=None,
14
  )
15
 
 
7
  from prompts import prompt_templates
8
 
9
  model = InferenceClientModel(
10
+ max_tokens=2000,
11
+ temperature=0,
12
+ model_id='gemini-2.0-flash',
13
  custom_role_conversions=None,
14
  )
15
 
app.py CHANGED
@@ -4,7 +4,7 @@ import requests
4
  import inspect
5
  import pandas as pd
6
  from agent import manager_agent
7
-
8
  # --- Constants ---
9
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
10
 
@@ -86,6 +86,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
86
  except Exception as e:
87
  print(f"Error running agent on task {task_id}: {e}")
88
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
 
89
 
90
  if not answers_payload:
91
  print("Agent did not produce any answers to submit.")
 
4
  import inspect
5
  import pandas as pd
6
  from agent import manager_agent
7
+ from time import sleep
8
  # --- Constants ---
9
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
10
 
 
86
  except Exception as e:
87
  print(f"Error running agent on task {task_id}: {e}")
88
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
89
+ sleep(30)
90
 
91
  if not answers_payload:
92
  print("Agent did not produce any answers to submit.")