Update app.py

app.py

@@ -17,10 +17,12 @@ from google.api_core import exceptions
 
 # 1. Initialize Client WITHOUT api_version
 # This specifically fixes the 'INVALID_ARGUMENT' error you received
-client = genai.Client(api_key=os.environ.get("GOOGLE_API_KEY"))
 
-# Use the direct model ID
-STABLE_MODEL = "gemini-1.5-flash"
+# Initialize the client with explicit API version 'v1'
+client = genai.Client(
+    api_key=userdata.get('GOOGLE_API_KEY'),
+    http_options=types.HttpOptions(api_version='v1') 
+)
 
 # --- Data Preparation ---
 df_credit = pd.DataFrame({'ID': [1111, 2222, 3333, 4444, 5555], 'Credit_Score': [455, 685, 825, 840, 350]})