flagtech commited on
Commit
e575683
·
verified ·
1 Parent(s): 23d4e93

update models list

Browse files
Files changed (1) hide show
  1. app.py +20 -4
app.py CHANGED
@@ -2,6 +2,25 @@ from google import genai
2
  import gradio as gr
3
  import os
4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  color_map = {
6
  '3': '#ccbfee',
7
  '2': '#beedc6',
@@ -40,10 +59,7 @@ web_tokenizer = gr.Interface(
40
  lines=6),
41
  gr.Dropdown(
42
  label='模型名稱',
43
- choices = [
44
- 'gemini-3-pro-preview',
45
- 'gemini-2.5-flash',
46
- ],
47
  value='gemini-3-pro-preview')
48
  ],
49
  outputs = [
 
2
  import gradio as gr
3
  import os
4
 
5
+ models = [
6
+ 'gemini-2.5-flash',
7
+ 'gemini-2.5-pro',
8
+ 'gemini-2.0-flash',
9
+ 'gemini-2.0-flash-lite',
10
+ 'gemma-3-1b-it',
11
+ 'gemma-3-4b-it',
12
+ 'gemma-3-12b-it',
13
+ 'gemma-3-27b-it',
14
+ 'gemma-3n-e4b-it',
15
+ 'gemma-3n-e2b-it',
16
+ 'gemini-pro-latest',
17
+ 'gemini-2.5-flash-lite',
18
+ 'gemini-2.5-flash-image',
19
+ 'gemini-3-pro-preview',
20
+ 'gemini-3-flash-preview',
21
+ 'gemini-3-pro-image-preview',
22
+ ]
23
+
24
  color_map = {
25
  '3': '#ccbfee',
26
  '2': '#beedc6',
 
59
  lines=6),
60
  gr.Dropdown(
61
  label='模型名稱',
62
+ choices = models,
 
 
 
63
  value='gemini-3-pro-preview')
64
  ],
65
  outputs = [