Moonfanz commited on
Commit
0208b24
·
verified ·
1 Parent(s): 3edcf79

Upload 4 files

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -71,7 +71,7 @@ safety_settings_g2 = [
71
  "threshold": "OFF"
72
  },
73
  {
74
- "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
75
  "threshold": "OFF"
76
  }
77
  ]
@@ -335,7 +335,8 @@ def chat_completions():
335
  gen_model = genai.GenerativeModel(
336
  model_name=model,
337
  generation_config=generation_config,
338
- safety_settings=safety_settings_g2 if model == 'gemini-2.0-flash-exp' else safety_settings,
 
339
  system_instruction=system_instruction
340
  )
341
 
 
71
  "threshold": "OFF"
72
  },
73
  {
74
+ "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
75
  "threshold": "OFF"
76
  }
77
  ]
 
335
  gen_model = genai.GenerativeModel(
336
  model_name=model,
337
  generation_config=generation_config,
338
+ # safety_settings=safety_settings_g2 if model == 'gemini-2.0-flash-exp' else safety_settings,
339
+ safety_settings=safety_settings,
340
  system_instruction=system_instruction
341
  )
342