qunfei commited on
Commit
4a4eae5
Β·
1 Parent(s): 60dbf47

adding key

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from io import StringIO
2
  import streamlit as st
 
3
  from loguru import logger
4
  from client import OpenAIService, SQLService, FacebookLLAMAService, GoogleT5Service
5
 
@@ -17,7 +18,7 @@ models = ['openai-text-david-003', 'google-t5', 'facebook-llama']
17
  st.sidebar.header('πŸŽƒ A SQL Transformer for Migration')
18
 
19
  model = st.sidebar.selectbox(label='Model', options=models, index=0)
20
- openai_key = st.sidebar.text_input("Openai Key", placeholder="No Key, No translate")
21
 
22
  source_database = st.sidebar.selectbox(
23
  label='πŸ“• Source Database',
 
1
  from io import StringIO
2
  import streamlit as st
3
+ import os
4
  from loguru import logger
5
  from client import OpenAIService, SQLService, FacebookLLAMAService, GoogleT5Service
6
 
 
18
  st.sidebar.header('πŸŽƒ A SQL Transformer for Migration')
19
 
20
  model = st.sidebar.selectbox(label='Model', options=models, index=0)
21
+ openai_key = os.environ.get("OPEN-KEY", None)
22
 
23
  source_database = st.sidebar.selectbox(
24
  label='πŸ“• Source Database',