Update chat_interface.py
Browse files- chat_interface.py +3 -1
chat_interface.py
CHANGED
@@ -8,12 +8,14 @@ import gradio as gr
|
|
8 |
from dotenv import load_dotenv
|
9 |
from openai import OpenAI
|
10 |
|
11 |
-
from langchain_community.embeddings import
|
|
|
12 |
from langchain_community.vectorstores import Chroma
|
13 |
|
14 |
from huggingface_hub import CommitScheduler
|
15 |
from pathlib import Path
|
16 |
|
|
|
17 |
|
18 |
load_dotenv()
|
19 |
|
|
|
8 |
from dotenv import load_dotenv
|
9 |
from openai import OpenAI
|
10 |
|
11 |
+
from langchain_community.embeddings.sentence_transformer import SentenceTransformerEmbeddings
|
12 |
+
|
13 |
from langchain_community.vectorstores import Chroma
|
14 |
|
15 |
from huggingface_hub import CommitScheduler
|
16 |
from pathlib import Path
|
17 |
|
18 |
+
embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-small')
|
19 |
|
20 |
load_dotenv()
|
21 |
|