Update app.py

app.py

@@ -10,7 +10,7 @@ import re
 from langchain.docstore.document import Document
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.retrievers import BM25Retriever
-from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool , Tool ,LiteLLMModel
 import gradio as gr
 import logging
 from nltk.corpus import words
@@ -174,7 +174,7 @@ When answering:
 """
 
 # Define the agent using smolagents
-model = HfApiModel("groq/llama3-8b-8192")  # Ensure the model is available
+model = LiteLLMModel("groq/llama3-8b-8192")  # Ensure the model is available
 agent = CodeAgent(
     tools=[retriever_tool, duckduckgo_search_tool], model=model, max_iterations=4, verbose=True
 )