fixing

app.py

@@ -1,9 +1,10 @@
 import os
+from transformers import pipeline
 import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, HfApiModel
+from smolagents import CodeAgent
 from smolagents.tools import DuckDuckGoSearchTool, PythonInterpreterTool
 import json
 import tempfile
@@ -13,6 +14,21 @@ from pathlib import Path
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+
+
+class HfApiModel:
+    """
+    Simple wrapper for Hugging Face pipeline as a replacement for smolagents.HfApiModel
+    """
+    def __init__(self, model_id: str, token: str = None):
+        self.model_id = model_id
+        self.token = token or os.getenv("HUGGINGFACE_INFERENCE_TOKEN")
+        self.pipe = pipeline("text-generation", model=model_id, token=self.token)
+
+    def __call__(self, prompt: str) -> str:
+        outputs = self.pipe(prompt, max_new_tokens=512, do_sample=True)
+        return outputs[0]["generated_text"]
+
 # --- Custom Tools ---
 class SerperSearchTool:
     """Enhanced search tool using Serper API for more reliable results"""