Update app.py

app.py

@@ -1,5 +1,6 @@
 # Import the necessary modules
 import google.generativeai as genai
+# Import the streamlit module and assign it to the alias st
 import streamlit as st
 
 # Configure the google-generativeai library by providing your API key
@@ -32,12 +33,13 @@ safety_settings = [
   }
 ]
 
-# Initialize the chat history and the model instance using session state
+model = genai.GenerativeModel(model_name="gemini-pro",
+                              generation_config=generation_config,
+                              safety_settings=safety_settings)
+
+# Initialize the chat history using session state
 # Use the get function to avoid KeyError if the key does not exist
 chat_history = st.session_state.get("chat_history", [])
-model = st.session_state.get("model", genai.GenerativeModel(model_name="gemini-pro",
-                              generation_config=generation_config,
-                              safety_settings=safety_settings))
 
 # Create a streamlit interface with a title, a description, and a chat container
 st.title("Gemini API") # the title of the interface