KitHung commited on
Commit
2e790d3
·
1 Parent(s): 359f615

Update: v1.11

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,6 @@
1
  import streamlit as st
 
 
2
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
3
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
4
  from llama_index.legacy.callbacks import CallbackManager
@@ -10,7 +12,7 @@ callback_manager = CallbackManager()
10
 
11
  api_base_url = "https://api.siliconflow.cn/v1"
12
  model = "internlm/internlm2_5-7b-chat"
13
- api_key = st.secrets["API_KEY"]
14
 
15
  llm =OpenAILike(model=model, api_base=api_base_url, api_key=api_key, is_chat_model=True, callback_manager=callback_manager)
16
 
 
1
  import streamlit as st
2
+ import os
3
+
4
  from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
5
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
6
  from llama_index.legacy.callbacks import CallbackManager
 
12
 
13
  api_base_url = "https://api.siliconflow.cn/v1"
14
  model = "internlm/internlm2_5-7b-chat"
15
+ api_key = os.environ.get("API_KEY")
16
 
17
  llm =OpenAILike(model=model, api_base=api_base_url, api_key=api_key, is_chat_model=True, callback_manager=callback_manager)
18