bstraehle commited on
Commit
5c795c7
·
1 Parent(s): 1311489

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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": 250,
17
- "top_p": 0.999,
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)