Update app.py

app.py

@@ -31,7 +31,14 @@ if password == st.secrets["real_password"]:
 
     with st.sidebar:
         st.title("Gemini Pro")
-
+        
+        CONFIG = {
+            "temperature": 0.5,
+            "top_p": 1,
+            "top_k": 32,
+            "max_output_tokens": 4096,
+        }
+        
         genai.configure(api_key=st.secrets["api_key"])
 
         uploaded_image = st.file_uploader(
@@ -48,7 +55,7 @@ if password == st.secrets["real_password"]:
 
     def get_response(messages, model="gemini-pro"):
         try:
-            model = genai.GenerativeModel(model)
+            model = genai.GenerativeModel(model, generation_config=genai.GenerationConfig(candidate_count=1, max_output_tokens=4096, temperature=0.6))
             res = model.generate_content(messages, stream=True, safety_settings=safety_settings)
             return res
         except: