Update app.py
Browse files
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 |
-
|
| 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=
|
| 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
|