arittrabag commited on
Commit
432d854
·
verified ·
1 Parent(s): 8449547

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -17,6 +17,8 @@ GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
17
  if not GEMINI_API_KEY:
18
  raise RuntimeError("GEMINI_API_KEY not found in environment.")
19
 
 
 
20
  def call_gemini(prompt: str) -> str:
21
  """Use Gemini via REST API instead of gRPC-based SDK"""
22
  url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}"
 
17
  if not GEMINI_API_KEY:
18
  raise RuntimeError("GEMINI_API_KEY not found in environment.")
19
 
20
+ model = genai.GenerativeModel("gemini-2.5-flash")
21
+
22
  def call_gemini(prompt: str) -> str:
23
  """Use Gemini via REST API instead of gRPC-based SDK"""
24
  url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}"