SorovotPelo commited on
Commit
fda1459
·
verified ·
1 Parent(s): 8ba2d64

Update stream_videos.py

Browse files
Files changed (1) hide show
  1. stream_videos.py +2 -2
stream_videos.py CHANGED
@@ -45,14 +45,14 @@ safety_settings = [
45
  ]
46
 
47
  text_model = genai.GenerativeModel(
48
- model_name="gemini-1.5-flash",
49
  generation_config=text_generation_config,
50
  safety_settings=safety_settings,
51
  system_instruction=system_prompt
52
  )
53
 
54
  image_model = genai.GenerativeModel(
55
- model_name="gemini-1.5-flash",
56
  generation_config=image_generation_config,
57
  safety_settings=safety_settings,
58
  system_instruction=image_prompt
 
45
  ]
46
 
47
  text_model = genai.GenerativeModel(
48
+ model_name="gemini-2.0-flash-thinking-exp-01-21",
49
  generation_config=text_generation_config,
50
  safety_settings=safety_settings,
51
  system_instruction=system_prompt
52
  )
53
 
54
  image_model = genai.GenerativeModel(
55
+ model_name="gemini-2.0-flash-thinking-exp-01-21",
56
  generation_config=image_generation_config,
57
  safety_settings=safety_settings,
58
  system_instruction=image_prompt