Update app.py

app.py

@@ -14,8 +14,39 @@ app.config['JSON_AS_ASCII'] = False
 CORS(app, resources={r"/*": {"origins": "*"}})
 
 API_KEY = os.environ['API_KEY']
+
 genai.configure(api_key=API_KEY)
-model = genai.GenerativeModel('gemini-pro')
+
+safety_settings = [
+    {
+        "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"
+    },
+]
+
+model = genai.GenerativeModel(
+    model_name = 'gemini-pro'
+    safety_settings = safety_settings,
+    generation_config = genai.GenerationConfig(
+    candidate_count = 1,
+    max_output_tokens = 4096,
+    temperature = 0.95,
+    top_p = 0.9,
+    top_k = 40
+)
+)
 
 def get_response(prompt:str):
     try: