Update app.py

app.py

@@ -11,13 +11,13 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, InferenceClientModel, LiteLLModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, InferenceClientModel, LiteLLMModel
 
 model_id = "gemini/gemini-2.0-flash"
 
 #agent = CodeAgent(tools=[], model=HfApiModel(token = os.getenv('TT_tokenik')), add_base_tools=True)
 
-agent = CodeAgent(tools=[], model=LiteLLModel(model_id=model_id, token = os.getenv('TT_tokenik')), add_base_tools=True)
+agent = CodeAgent(tools=[], model=LiteLLMModel(model_id=model_id, token = os.getenv('TT_tokenik')), add_base_tools=True)
 
 #agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())