Spaces:
Running
Running
Upload 4 files
Browse files
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 |
|