Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from smolagents import CodeAegnt,LiteLLMModel, DuckDuckGoSearchTool
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -19,26 +19,21 @@ If you are asked for a comma separated list, apply the above rules depending of
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        self.model = InferenceClientModel(
-            model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-            token=os.getenv("HF_TOKEN")
+        self.model = LiteLLMModel(
+            model_id="groq/llama-3.3-70b-versatile",
+            api_key=os.getenv("GROQ_API_KEY")
         )
         self.agent = CodeAgent(
             tools=[DuckDuckGoSearchTool()],
-            model=self.model
+            model=self.model,
+            max_steps=8
         )
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         full_question = f"{SYSTEM_PROMPT}\n\nQuestion: {question}"
         raw_answer = self.agent.run(full_question)
 
-        # Extrait uniquement ce qui suit "FINAL ANSWER:"
-        raw_answer = str(raw_answer)
-        if "FINAL ANSWER:" in raw_answer:
-            answer = raw_answer.split("FINAL ANSWER:")[-1].strip()
-        else:
-            answer = raw_answer.strip()
-
+        answer = raw_answer.split("FINAL ANSWER:")[-1].strip() if "FINAL ANSWER:" in raw_answer else raw_answer.strip()
         print(f"Agent returning answer: {answer}")
         return answer