Update app.py

app.py

@@ -2,17 +2,16 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool
-from transformers import pipeline
+from smolagents import CodeAgent, DuckDuckGoSearchTool,TransformersModel
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-pipe = pipeline("text-generation", model="openai-community/gpt2")
+
 
 # --- Define Agent ---
 class SmolAgentWrapper:
     def __init__(self):
-        self.model = pipe  # Uses flan-t5-base via HF Inference API
+        self.model = TransformersModel(model_id="gpt2")  # Uses flan-t5-base via HF Inference API
         self.tools = [DuckDuckGoSearchTool()]  # You can add more tools here
         self.agent = CodeAgent(model=self.model, tools=self.tools)