Update app.py

app.py

@@ -6,9 +6,16 @@ import os
 import google.generativeai as genai
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+generation_config = {
+  "temperature": 0,
+  "top_p": 1,
+  "top_k": 1,
+  "max_output_tokens": 2048,
+}
 
 ## function to load Gemini Pro model and get repsonses
-model=genai.GenerativeModel("gemini-pro") 
+model = genai.GenerativeModel(model_name="gemini-pro",
+                              generation_config=generation_config) 
 chat = model.start_chat(history=[])
 def get_gemini_response(question):