BinduRP commited on
Commit
c9f040a
·
verified ·
1 Parent(s): e5fd82e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ from google.api_core import exceptions
32
  client = genai.Client(api_key=os.environ.get('GOOGLE_API_KEY'))
33
 
34
  # Use gemini-1.5-flash for the best free-tier stability
35
- FREE_STABLE_MODEL = "gemini-flash-latest"
36
 
37
  # --- Data Preparation ---
38
  df_credit = pd.DataFrame({'ID': [1111, 2222, 3333, 4444, 5555], 'Credit_Score': [455, 685, 825, 840, 350]})
@@ -67,7 +67,7 @@ def process_loan(applicant_id, optional_name):
67
 
68
  # Using direct file objects in the contents list
69
  response = client.models.generate_content(
70
- model=FREE_STABLE_MODEL,
71
  contents=[risk_policy, rate_policy, prompt]
72
  )
73
  return response.text
 
32
  client = genai.Client(api_key=os.environ.get('GOOGLE_API_KEY'))
33
 
34
  # Use gemini-1.5-flash for the best free-tier stability
35
+ STABLE_MODEL = "gemini-flash-latest"
36
 
37
  # --- Data Preparation ---
38
  df_credit = pd.DataFrame({'ID': [1111, 2222, 3333, 4444, 5555], 'Credit_Score': [455, 685, 825, 840, 350]})
 
67
 
68
  # Using direct file objects in the contents list
69
  response = client.models.generate_content(
70
+ model=STABLE_MODEL,
71
  contents=[risk_policy, rate_policy, prompt]
72
  )
73
  return response.text