Update app.py
Browse files
app.py
CHANGED
@@ -12,9 +12,9 @@ config = {
|
|
12 |
"max_output_tokens": 800,
|
13 |
#"model": "text-bison@001",
|
14 |
"model": "gemini-pro",
|
15 |
-
"temperature": 0,
|
16 |
-
"top_k":
|
17 |
-
"top_p": 0
|
18 |
}
|
19 |
|
20 |
wandb.login(key = wandb_api_key)
|
|
|
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,
|
18 |
}
|
19 |
|
20 |
wandb.login(key = wandb_api_key)
|