Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from crewai import Agent, Task, Crew
+import google.generativeai as genai
 from langchain_google_genai import ChatGoogleGenerativeAI
 from dotenv import load_dotenv
 import os
@@ -9,7 +10,8 @@ load_dotenv()
 
 # Initialize Google Gemini AI
 llm = ChatGoogleGenerativeAI(
-    api_key=os.getenv('GOOGLE_API_KEY')
+    api_key=os.getenv('GOOGLE_API_KEY'),
+    model = genai.GenerativeModel('gemini-pro')# Replace with the correct model name
 )
 
 # Define Agents