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, DuckDuckGoSearchTool, FinalAnswerTool, InferenceClientModel, Tool, tool, VisitWebpageTool, HfApiModel, LiteLLMModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, PythonInterpreterTool, Tool, tool, VisitWebpageTool, LiteLLMModel
 from groq import Groq
 
 # (Keep Constants as is)
@@ -17,14 +17,14 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.model = LiteLLMModel(
-            "openai/deepseek-r1-distill-llama-70b",
+            "llama-3.1-8b-instant",
             api_base="https://api.groq.com/openai/v1",
             api_key=GROQ_KEY
         )
         self.model.flatten_messages_as_text = True
 
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool(), FinalAnswerTool(), VisitWebpageTool()],
+            tools=[DuckDuckGoSearchTool(), FinalAnswerTool(), VisitWebpageTool(), PythonInterpreterTool(), ],
             model=self.model,
             add_base_tools=True,
             verbosity_level=2