Update app.py

app.py

@@ -10,36 +10,7 @@ from time import time as t
 
 app = Flask(__name__)
 
-generation_config = {
- "temperature": 0.7,
- "top_p": 1,
- "top_k": 1,
- "max_output_tokens": 300,
-}
-
-safety_settings = [
- {
-    "category": "HARM_CATEGORY_HARASSMENT",
-    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
- },
- {
-    "category": "HARM_CATEGORY_HATE_SPEECH",
-    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
- },
- {
-    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
- },
- {
-    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-    "threshold": "BLOCK_MEDIUM_AND_ABOVE"
- },
-]
-
-model = genai.GenerativeModel(
- model_name="gemini-pro",
- generation_config=generation_config,
- safety_settings=safety_settings)
+
 
 
 
@@ -121,7 +92,37 @@ def IMGEN():
 @app.route('/generativeai', methods=['POST'])
 def Genration():
     import google.generativeai as genai
-    global model,safety_settings,generation_config
+    generation_config = {
+     "temperature": 0.7,
+     "top_p": 1,
+     "top_k": 1,
+     "max_output_tokens": 300,
+    }
+    
+    safety_settings = [
+     {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+     },
+     {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+     },
+     {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+     },
+     {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
+     },
+    ]
+    
+    model = genai.GenerativeModel(
+     model_name="gemini-pro",
+     generation_config=generation_config,
+     safety_settings=safety_settings)
+
     data = request.json
     prompt = data.get('prompt', '')
     messages = data.get('messages', [])