pgurazada1 commited on
Commit
e7239b1
·
verified ·
1 Parent(s): ab78970

Update chat_interface.py

Browse files
Files changed (1) hide show
  1. 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 AnyscaleEmbeddings
 
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