had Litellm update to include gemini model

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, LiteLLMModel
 import datetime
 import requests
 import pytz
@@ -37,12 +37,13 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 
 final_answer = FinalAnswerTool()
-model = HfApiModel(
-token = os.environ["HF_TOKEN"],
-max_tokens=2096,
-temperature=0.5,
-model_id= os.environ['MODEL'],# it is possible that this model may be overloaded
-custom_role_conversions=None,
+
+model = LiteLLMModel(
+    token = os.environ["HF_TOKEN"],
+    max_tokens=2096,
+    temperature=0.5,
+    model_id= os.environ['MODEL'],# it is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
 @tool