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