Spaces:

airworkx
/

gemini

Build error

gemini / utils.py

Update utils.py

a505ab1 verified over 2 years ago

846 Bytes

# Original code
SAFETY_SETTTINGS = [
    {
        "category": "HARM_CATEGORY_SEXUAL",
        "threshold": "BLOCK_NONE",
    },
    {
        "category": "HARM_CATEGORY_DANGEROUS",
        "threshold": "BLOCK_NONE",
    },
    {
        "category": "HARM_CATEGORY_HARASSMENT",
        "threshold": "BLOCK_NONE",
    },
    {
        "category": "HARM_CATEGORY_HATE_SPEECH",
        "threshold": "BLOCK_NONE",
    },
    {
        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
        "threshold": "BLOCK_NONE",
    },
    {
        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
        "threshold": "BLOCK_NONE",
    },
]

# Modified code to allow uninterrupted chat session
MODIFIED_SAFETY_SETTTINGS = []

# Set all thresholds to "ALLOW_ALL" to disable blocking
for setting in MODIFIED_SAFETY_SETTTINGS:
    setting["threshold"] = "ALLOW_ALL"

	# Original code
	SAFETY_SETTTINGS = [
	{
	"category": "HARM_CATEGORY_SEXUAL",
	"threshold": "BLOCK_NONE",
	},
	{
	"category": "HARM_CATEGORY_DANGEROUS",
	"threshold": "BLOCK_NONE",
	},
	{
	"category": "HARM_CATEGORY_HARASSMENT",
	"threshold": "BLOCK_NONE",
	},
	{
	"category": "HARM_CATEGORY_HATE_SPEECH",
	"threshold": "BLOCK_NONE",
	},
	{
	"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
	"threshold": "BLOCK_NONE",
	},
	{
	"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
	"threshold": "BLOCK_NONE",
	},
	]

	# Modified code to allow uninterrupted chat session
	MODIFIED_SAFETY_SETTTINGS = []

	# Set all thresholds to "ALLOW_ALL" to disable blocking
	for setting in MODIFIED_SAFETY_SETTTINGS:
	setting["threshold"] = "ALLOW_ALL"