AKKI-AFK commited on
Commit
af7bdb7
·
verified ·
1 Parent(s): 40be589

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -28,7 +28,7 @@ def process_loan_data(df: pd.DataFrame):
28
 
29
  def get_gemini_decision(segment, pd_val, lgd_val, ead_val, ecl_val):
30
  """Ask Gemini to decide and sanitize any invalid JSON output."""
31
- model = genai.GenerativeModel("gemini-1.5-pro")
32
  system_prompt = """You are a financial risk advisor.
33
  Return only valid JSON in this exact schema:
34
  {"action":"increase_interest"|"reduce_disbursement"|"maintain","rationale":"string","confidence":float}
 
28
 
29
  def get_gemini_decision(segment, pd_val, lgd_val, ead_val, ecl_val):
30
  """Ask Gemini to decide and sanitize any invalid JSON output."""
31
+ model = genai.GenerativeModel("gemini-2.0-flash-lite")
32
  system_prompt = """You are a financial risk advisor.
33
  Return only valid JSON in this exact schema:
34
  {"action":"increase_interest"|"reduce_disbursement"|"maintain","rationale":"string","confidence":float}