Spaces:

rairo
/

stmt-api

Running

rairo commited on Apr 20

Commit

179236b

verified ·

1 Parent(s): b61acc0

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -38,7 +38,7 @@ def configure_gemini(api_key):
     """Configure Gemini AI model."""
     try:
         genai.configure(api_key=api_key)
-        return genai.GenerativeModel('gemini-2.0-flash')
     except Exception as e:
         logging.error(f"Error configuring Gemini: {str(e)}")
         raise

     """Configure Gemini AI model."""
     try:
         genai.configure(api_key=api_key)
+        return genai.GenerativeModel('gemini-2.5-flash')
     except Exception as e:
         logging.error(f"Error configuring Gemini: {str(e)}")
         raise