degbu embeddings
Browse files
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 |
|