bstraehle commited on
Commit
3c5b4aa
·
verified ·
1 Parent(s): edf6a1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,7 +9,7 @@ project = os.environ["PROJECT"]
9
 
10
  config = {
11
  "max_output_tokens": 1000,
12
- "model": "gemini-pro",
13
  "temperature": 0.1,
14
  "top_k": 40,
15
  "top_p": 1.0,
@@ -35,7 +35,7 @@ def invoke(prompt):
35
  if (prompt == ""):
36
  raise gr.Error("Prompt is required.")
37
 
38
- raise gr.Error("Clone and bring your own credentials.")
39
 
40
  completion = ""
41
 
@@ -57,7 +57,7 @@ def invoke(prompt):
57
  return completion
58
 
59
  description = """<a href='https://www.gradio.app/'>Gradio</a> UI using the <a href='https://cloud.google.com/vertex-ai'>Google Vertex AI</a> API
60
- with Gemini Pro model."""
61
 
62
  gr.close_all()
63
 
 
9
 
10
  config = {
11
  "max_output_tokens": 1000,
12
+ "model": "gemini-1.5-pro-preview-0514",
13
  "temperature": 0.1,
14
  "top_k": 40,
15
  "top_p": 1.0,
 
35
  if (prompt == ""):
36
  raise gr.Error("Prompt is required.")
37
 
38
+ #raise gr.Error("Clone and bring your own credentials.")
39
 
40
  completion = ""
41
 
 
57
  return completion
58
 
59
  description = """<a href='https://www.gradio.app/'>Gradio</a> UI using the <a href='https://cloud.google.com/vertex-ai'>Google Vertex AI</a> API
60
+ with Gemini 1.5 Pro model."""
61
 
62
  gr.close_all()
63