Upload app.py

app.py

@@ -5,7 +5,7 @@ import inspect
 import pandas as pd
 
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool
+from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, InferenceClientModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -15,10 +15,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        grok_model = pipeline("text-generation", model="distilgpt2")
         openAI_model = OpenAIServerModel(model_id="gpt-4o-mini", api_key="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")
         self.agent = CodeAgent(
-            model = grok_model,
+            model = InferenceClientModel(),
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             add_base_tools = True,
         )