Update app.py

app.py

@@ -21,14 +21,19 @@ GEMINI_API_KEY = os.getenv("gemini_api")
 
 genai.configure(api_key=GEMINI_API_KEY)
 generation_config = {
-    "temperature": 1,
-    "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 8192,
+  "temperature": 1,
+  "top_p": 0.95,
+  "top_k": 64,
+  "max_output_tokens": 8192,
+  "response_mime_type": "text/plain",
 }
 
-model = genai.GenerativeModel("gemini-pro", generation_config=generation_config)
-chat_model = genai.GenerativeModel("gemini-pro")
+model = genai.GenerativeModel(
+  model_name="gemini-2.0-pro-exp-02-05",
+  generation_config=generation_config,
+)
+
+chat_model = genai.GenerativeModel('"gemini-2.0-pro-exp-02-05"')
 
 # Create and save a simple model on first run
 def create_initial_model():