Prathamesh1420 commited on
Commit
255ea39
·
verified ·
1 Parent(s): 678be94

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import streamlit as st
2
  from crewai import Agent, Task, Crew
3
- import google.generativeai as genai
4
  from langchain_google_genai import ChatGoogleGenerativeAI
5
  from dotenv import load_dotenv
6
  import os
@@ -11,7 +10,7 @@ load_dotenv()
11
  # Initialize Google Gemini AI
12
  llm = ChatGoogleGenerativeAI(
13
  api_key=os.getenv('GOOGLE_API_KEY'),
14
- model = genai.GenerativeModel('gemini-pro')# Replace with the correct model name
15
  )
16
 
17
  # Define Agents
 
1
  import streamlit as st
2
  from crewai import Agent, Task, Crew
 
3
  from langchain_google_genai import ChatGoogleGenerativeAI
4
  from dotenv import load_dotenv
5
  import os
 
10
  # Initialize Google Gemini AI
11
  llm = ChatGoogleGenerativeAI(
12
  api_key=os.getenv('GOOGLE_API_KEY'),
13
+ model="models/gemini-pro" # Replace with the correct model name
14
  )
15
 
16
  # Define Agents