Update streamlit_app.py

streamlit_app.py

@@ -62,18 +62,18 @@ 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" }, ]
 
 #Gemini Model
 model_id = expander.text_input("Gemini Model", "gemini-2.0-flash-exp")
+# Returns the "context window" for the model,
+# which is the combined input and output token limits.
+    print(f"{model_id.input_token_limit=}")
+    print(f"{model_id.output_token_limit=}")
+# ( input_token_limit=30720, output_token_limit=2048 )
+# [END tokens_context_window]
 
 # Analysis prompt
 analysis_prompt = expander.text_area("Enter analysis")