Update app.py

app.py

@@ -7,7 +7,7 @@ from typing import List
 
 from langchain.text_splitter import CharacterTextSplitter
 from tools.final_answer import FinalAnswerTool
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from Gradio_UI import GradioUI
 
 # === TOOLS ===
@@ -72,12 +72,11 @@ model_config = agent_config["model"]["data"]
 # === BUILD MODEL ===
 from smolagents import HfApiModel
 
-model = HfApiModel(
-    model_id="google/gemma-2b-it",
-    token=os.getenv("HF_TOKEN"),
+model = LiteLLMModel(
+    model_id="gemini/gemini-2.0-flash-lite",
+    api_key=os.getenv("GEMINI_API_KEY"),
     temperature=0.5,
     max_tokens=1024,
-    custom_role_conversions=None,
 )
 
 model.last_input_token_count = 0