Update main.py

main.py

@@ -1,8 +1,8 @@
 import streamlit as st
 from vanna.remote import VannaDefault
-
-# Set your Vanna API key and model name from Streamlit secrets
-api_key = st.secrets["general"]["vanna_api_key"]
+import os
+# Set your Vanna API key 
+api_key = os.environ.get("VANNA_API_KEY")
 vanna_model_name = 'chinook'  # Assuming 'chinook' is your model name
 
 # Initialize Vanna