PRIYANSHUDHAKED commited on
Commit
c7ba6ae
·
verified ·
1 Parent(s): 830e830

Update qa_engine.py

Browse files
Files changed (1) hide show
  1. qa_engine.py +6 -2
qa_engine.py CHANGED
@@ -2,10 +2,14 @@ import google.generativeai as genai
2
  import os
3
 
4
  # Configure the Gemini API
5
- genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 
 
 
6
 
7
  # Initialize the Gemini Pro model for text-based tasks
8
- model = genai.GenerativeModel('gemini-pro')
9
 
10
  def get_answer(question, video_data):
11
  prompt = f"""
 
2
  import os
3
 
4
  # Configure the Gemini API
5
+ import os
6
+ import google.generativeai as genai
7
+
8
+ os.environ["GOOGLE_API_KEY"] = "AIzaSyD0GxR2J1JxGic807Cc89Jq6MB4aDJYgDc"
9
+ genai.configure()
10
 
11
  # Initialize the Gemini Pro model for text-based tasks
12
+ model = genai.GenerativeModel('gemini-1.5-flash')
13
 
14
  def get_answer(question, video_data):
15
  prompt = f"""