update the interface

README.md

@@ -28,7 +28,7 @@ streamlit run app.py
 3. Update the `os.environ["OPENAI_API_KEY"] ` in the `train.py`
 4. Run `python3 train.py`
 
-The training will take 15 seconds, and cost around $2.00
+The training will take 15 seconds, and cost around $0.20
 ```
 chromadb.db.duckdb: loaded in 236 embeddings
 chromadb.db.duckdb: loaded in 1 collections

app.py

@@ -13,12 +13,14 @@ from langchain.callbacks import get_openai_callback
 # variables
 db_folder = "db"
 mode = "infer"  # "train" or "infer"
-
+st.title('DataHub QA Chat Demo')
 # set your OpenAI API key
 api_key = st.text_input('OPENAI API KEY')
 if api_key:
     os.environ["OPENAI_API_KEY"] = api_key
-
+else:
+    st.write("Please provide your OpenAI API KEY, the query is cheap ($0.0001 per query))")
+    exit()
 
 
 # initialize the language model