Update app.py

app.py

@@ -12,37 +12,7 @@ from time import time as t
 app = Flask(__name__)
 
 
-# Tracking API usage
-generation_config = {
- "temperature": 0.7,
- "top_p": 1,
- "top_k": 1,
- "max_output_tokens": 350,
-}
-
-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)
+
 
 
 @app.route('/mistral7b', methods=['POST'])
@@ -129,6 +99,37 @@ def Genration():
     key = data.get('key', '')
     
     C=t()
+    # Tracking API usage
+    generation_config = {
+     "temperature": 0.7,
+     "top_p": 1,
+     "top_k": 1,
+     "max_output_tokens": 350,
+    }
+    
+    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)
     genai.configure(api_key=key)
     response = model.generate_content(messages)