valentynliubchenko commited on
Commit
cfae62c
·
1 Parent(s): 3e3bb44

added gemini-2.0-flash-thinking-exp-01-21

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,10 +16,10 @@ from vertex_ai_service import VertexAIService
16
  load_dotenv()
17
  isFullVersion = os.getenv("COLLECTION_DATA_VERSION") != "True"
18
  if isFullVersion:
19
- model_names = ["gemini-1.5-flash", "gemini-1.5-pro", "gemini-flash-experimental", "gemini-pro-experimental", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp-1219",
20
  "gpt-4o-mini", "gpt-4o", "QR-processing"]
21
  else:
22
- model_names = ["gemini-1.5-flash", "gemini-1.5-pro", "gemini-flash-experimental", "gemini-pro-experimental", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp-1219", "QR-processing"]
23
 
24
  prompt_names = ["prompt_v1", "prompt_v2", "prompt_v3"]
25
  # example_list = [["./examples/" + example] for example in os.listdir("examples")]
 
16
  load_dotenv()
17
  isFullVersion = os.getenv("COLLECTION_DATA_VERSION") != "True"
18
  if isFullVersion:
19
+ model_names = ["gemini-1.5-flash", "gemini-1.5-pro", "gemini-flash-experimental", "gemini-pro-experimental", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp-01-21",
20
  "gpt-4o-mini", "gpt-4o", "QR-processing"]
21
  else:
22
+ model_names = ["gemini-1.5-flash", "gemini-1.5-pro", "gemini-flash-experimental", "gemini-pro-experimental", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp-01-21", "QR-processing"]
23
 
24
  prompt_names = ["prompt_v1", "prompt_v2", "prompt_v3"]
25
  # example_list = [["./examples/" + example] for example in os.listdir("examples")]