adilos1 commited on
Commit
d1e07aa
·
verified ·
1 Parent(s): 9cb9c16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -46,9 +46,10 @@ class MyAgent:
46
 
47
  self.model = LiteLLMModel(
48
  # model_id="gemini/gemini-2.0-flash-lite",
49
- model_id="openai/gpt-4o",
50
- # api_key=gemini_api_key,
51
- api_key=openai_api_key,
 
52
  system_prompt=SYSTEM_PROMPT
53
  )
54
 
 
46
 
47
  self.model = LiteLLMModel(
48
  # model_id="gemini/gemini-2.0-flash-lite",
49
+ # model_id="openai/gpt-4o",
50
+ model_id="gemini/gemini-2.5-flash-lite-preview-06-17",
51
+ api_key=gemini_api_key,
52
+ # api_key=openai_api_key,
53
  system_prompt=SYSTEM_PROMPT
54
  )
55