Update app.py

app.py

@@ -3,7 +3,7 @@ import streamlit as st
 from snowflake.snowpark import Session
 from snowflake.ml.utils.connection_params import SnowflakeLoginOptions
 from langchain_community.utilities import SQLDatabase
-from langchain_openai import OpenAI
+from langchain_llms import HuggingFaceHub
 from langchain.chains import create_sql_query_chain
 
 @st.cache_resource(show_spinner="Connecting...")
@@ -18,8 +18,8 @@ def getSession():
         + f"?warehouse={pars['warehouse']}&role={pars['role']}")
     db = SQLDatabase.from_uri(url)
 
-    openai_key = os.environ["OPENAI_API_KEY"]
-    llm = OpenAI(openai_api_key=openai_key)
+     os.environ["HUGGINGFACEHUB_API_TOKEN"] = ""
+    llm = HuggingFaceHub(repo_id= "ravithejakandi/Sf-Arctic-Demo-Enu")
     chain = create_sql_query_chain(llm, db)
     return session, db, chain