Update app.py

app.py

@@ -6,8 +6,7 @@ import subprocess
 import pandas as pd
 import torch, spaces
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-# from smolagents import CodeAgent, HfApiModel
-from smolagents.agent import LocalLLM, CodeAgent
+from smolagents import CodeAgent, HfApiModel
 from smolagents.message import MessageRole
 from huggingface_hub import InferenceClient, hf_hub_download
 
@@ -61,7 +60,7 @@ def check_token_access():
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent(LocalLLM):
+class BasicAgent:
     def __init__(self, model_id="meta-llama/Llama-3.1-8B-Instruct", hf_token=None):
         print("BasicAgent initialized.")
         print("ENV-HF_TOKEN-LEN", len(hf_token), file=sys.stderr)