Update main.py

main.py

@@ -17,7 +17,7 @@ if not st.session_state.api_key:
 # Configure generative model
 try:
     genai.configure(api_key=st.session_state.api_key)
-    model = genai.GenerativeModel(model_name="gemini-1.5-flash")
+    model = genai.GenerativeModel(model_name="gemini-1.5-flash-001")
 except Exception as e:
     st.error("Invalid API key or configuration error.")
     st.stop()