ll7098ll commited on
Commit
d23fd89
·
verified ·
1 Parent(s): e0d7db5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,12 +9,12 @@ genai.configure(api_key=os.environ["GEMINI_API_KEY"])
9
  generation_config = {
10
  "temperature": 1.0,
11
  "top_p": 0.95,
12
- "top_k": 64,
13
  "max_output_tokens": 7000,
14
  "response_mime_type": "text/plain",
15
  }
16
  model = genai.GenerativeModel(
17
- model_name="gemini-1.5-pro-exp-0827",
18
  generation_config=generation_config,
19
  )
20
 
 
9
  generation_config = {
10
  "temperature": 1.0,
11
  "top_p": 0.95,
12
+ "top_k": 40,
13
  "max_output_tokens": 7000,
14
  "response_mime_type": "text/plain",
15
  }
16
  model = genai.GenerativeModel(
17
+ model_name="gemini-1.5-pro-002",
18
  generation_config=generation_config,
19
  )
20