Mouhamedamar commited on
Commit
6076917
·
verified ·
1 Parent(s): 894c83f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -61,7 +61,7 @@ def analyze_youtube_video(url: str, question: str) -> str:
61
  return "Unable to perform analysis: Google API key not set. Get it from https://aistudio.google.com/"
62
 
63
  llm = ChatGoogleGenerativeAI(
64
- model="gemini-1.5-flash",
65
  google_api_key=api_key,
66
  temperature=0,
67
  max_output_tokens=4096
@@ -102,7 +102,7 @@ def analyze_text_content(content: str, question: str) -> str:
102
  return "Unable to perform analysis: Google API key not set."
103
 
104
  llm = ChatGoogleGenerativeAI(
105
- model="gemini-1.5-flash",
106
  google_api_key=api_key,
107
  temperature=0,
108
  max_output_tokens=4096
@@ -143,7 +143,7 @@ def direct_reasoning(question: str, context: str = "") -> str:
143
  return "Google API key not set."
144
 
145
  llm = ChatGoogleGenerativeAI(
146
- model="gemini-1.5-flash",
147
  google_api_key=api_key,
148
  temperature=0,
149
  max_output_tokens=4096
@@ -196,7 +196,7 @@ def get_llm():
196
  if not api_key:
197
  raise ValueError("GOOGLE_API_KEY environment variable not set. Get it from https://aistudio.google.com/")
198
  return ChatGoogleGenerativeAI(
199
- model="gemini-1.5-flash",
200
  google_api_key=api_key,
201
  temperature=0,
202
  max_output_tokens=4096
 
61
  return "Unable to perform analysis: Google API key not set. Get it from https://aistudio.google.com/"
62
 
63
  llm = ChatGoogleGenerativeAI(
64
+ model="gemini-2.5-flash",
65
  google_api_key=api_key,
66
  temperature=0,
67
  max_output_tokens=4096
 
102
  return "Unable to perform analysis: Google API key not set."
103
 
104
  llm = ChatGoogleGenerativeAI(
105
+ model="gemini-2.5-flash",
106
  google_api_key=api_key,
107
  temperature=0,
108
  max_output_tokens=4096
 
143
  return "Google API key not set."
144
 
145
  llm = ChatGoogleGenerativeAI(
146
+ model="gemini-2.5-flash",
147
  google_api_key=api_key,
148
  temperature=0,
149
  max_output_tokens=4096
 
196
  if not api_key:
197
  raise ValueError("GOOGLE_API_KEY environment variable not set. Get it from https://aistudio.google.com/")
198
  return ChatGoogleGenerativeAI(
199
+ model="gemini-2.5-flash",
200
  google_api_key=api_key,
201
  temperature=0,
202
  max_output_tokens=4096