Update app.py

app.py

@@ -10,7 +10,7 @@ from parler_tts import ParlerTTSForConditionalGeneration
 import soundfile as sf
 from llama_index.core.agent import ReActAgent
 from llama_index.core.tools import FunctionTool
-from llama_index.core.llms import GroqLLM
+from llama_index.llms.groq import Groq
 from PIL import Image
 from tavily import TavilyClient
 import requests
@@ -99,7 +99,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False):
         FunctionTool.from_defaults(fn=image_generation, name="Image Generation"),
     ]
 
-    llm = GroqLLM(model_name=MODEL)
+    llm = Groq(model_name=MODEL, api_key=os.environ.get("GROQ_API_KEY"))
     agent = ReActAgent.from_tools(tools, llm=llm, verbose=True)
 
     if image: