Rooni commited on
Commit
daec028
·
verified ·
1 Parent(s): 2e6be27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def generate(description, system_role_name, max_tokens):
29
  payload = {
30
  'messages': [{'role': 'system', 'content': system_role}, {'role': 'user', 'content': description}],
31
  'max_tokens': max_tokens,
32
- 'model': "gemini-1.5-pro-latest",
33
  'stream': True
34
  }
35
 
 
29
  payload = {
30
  'messages': [{'role': 'system', 'content': system_role}, {'role': 'user', 'content': description}],
31
  'max_tokens': max_tokens,
32
+ 'model': "gemini-2.0-flash",
33
  'stream': True
34
  }
35