patricksamuel commited on
Commit
40bf437
·
verified ·
1 Parent(s): c2e9bfb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import os
2
  import gradio as gr
3
  import inspect
@@ -147,7 +150,7 @@ Follow this cycle until you find the answer:
147
  visit_website
148
  ],
149
  model=LitellmModel(
150
- model="gemini/gemini-pro", # This tells LiteLLM to route to Gemini
151
  api_key=os.getenv("GEMINI_API_KEY")
152
  )
153
 
 
1
+ from __future__ import annotations
2
+
3
+
4
  import os
5
  import gradio as gr
6
  import inspect
 
150
  visit_website
151
  ],
152
  model=LitellmModel(
153
+ model="gemini/gemini-pro",
154
  api_key=os.getenv("GEMINI_API_KEY")
155
  )
156