Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ wandb_api_key = os.environ["WANDB_API_KEY"]
|
|
10 |
|
11 |
config = {
|
12 |
"max_output_tokens": 800,
|
13 |
-
|
14 |
-
"model": "gemini-pro",
|
15 |
"temperature": 0.1,
|
16 |
"top_k": 40,
|
17 |
"top_p": 1.0,
|
|
|
10 |
|
11 |
config = {
|
12 |
"max_output_tokens": 800,
|
13 |
+
"model": "text-bison@001",
|
14 |
+
#"model": "gemini-pro",
|
15 |
"temperature": 0.1,
|
16 |
"top_k": 40,
|
17 |
"top_p": 1.0,
|