Update streamlit_app.py

streamlit_app.py

@@ -62,7 +62,13 @@ expander = st.sidebar.expander("Options", expanded=True)
 API_KEY = expander.text_input("Google API Key", "AIzaSyB2uDj6IeuNszSuk80feW-eBgpIDH0WFSY")
 if API_KEY:
     genai.configure(api_key=API_KEY)
-    
+# Returns the "context window" for the model,
+# which is the combined input and output token limits.
+    print(f"{model_info.input_token_limit=}")
+    print(f"{model_info.output_token_limit=}")
+# ( input_token_limit=30720, output_token_limit=2048 )
+# [END tokens_context_window]
+
 #Safety Settings
 safety_settings = [ { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF" }, { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF" }, { "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF" }, { "category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF" }, ]