Update app.py

app.py

@@ -22,7 +22,8 @@ difficulty = 1  # 1 = Easy, 2 = Medium, 3 = Hard
 # --- Gemini Model ---
 #model = genai.GenerativeModel('gemini-pro')  # Or 'gemini-pro-vision' if you need image input
 #model = genai.GenerativeModel('gemini-1.5-pro-latest')
-model = genai.GenerativeModel('gemini-2-flash')
+#model = genai.GenerativeModel('gemini-2-flash')
+model = genai.GenerativeModel('gemini-pro', generate_response_clause=genai.types.GenerateResponseClause(model='gemini-pro'))
 
 
 # --- Functions ---