mahynski commited on
Commit
ebb60db
·
1 Parent(s): 19934cf

degbu embeddings

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ with st.sidebar:
94
  Settings.tokenizer = tiktoken.encoding_for_model(llm_name).encode
95
  Settings.num_output = max_tokens
96
  Settings.context_window = 4096 # max possible
97
- Settings.embed_model = OpenAIEmbedding()
98
  elif provider == 'huggingface':
99
  os.environ['HFTOKEN'] = str(llm_token)
100
 
 
94
  Settings.tokenizer = tiktoken.encoding_for_model(llm_name).encode
95
  Settings.num_output = max_tokens
96
  Settings.context_window = 4096 # max possible
97
+ Settings.embed_model = OpenAIEmbedding(api_key=os.environ.get("OPENAI_API_KEY"))
98
  elif provider == 'huggingface':
99
  os.environ['HFTOKEN'] = str(llm_token)
100