Add system prompt and reduce number of questions submitted for speed

agent.py

@@ -1,13 +1,17 @@
 from smolagents import (
     CodeAgent,
-    OpenAIServerModel,
+    InferenceClientModel,
 )
 
 class DRAgent:
     def __init__(self):
+        self.system_prompt = "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
+
         self.agent = CodeAgent(
             tools=[],
-            model = OpenAIServerModel(model_id="gpt-4o")
+            model=InferenceClientModel(
+            model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
+            ),
         )
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:

app.py

@@ -76,7 +76,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-    for item in questions_data:
+    for item in questions_data[:1]: # Reduce the number of submitted questions for speed
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None: