pgurazada1 commited on
Commit
04e742d
1 Parent(s): 1b055d9

Update chat_interface.py

Browse files
Files changed (1) hide show
  1. chat_interface.py +2 -2
chat_interface.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
 
6
  from openai import AzureOpenAI
7
 
8
- from langchain_community.embeddings.sentence_transformer import SentenceTransformerEmbeddings
9
  from langchain_community.vectorstores import Chroma
10
 
11
  from typing import List
@@ -54,7 +54,7 @@ client = AzureOpenAI(
54
  api_version = "2024-02-01"
55
  )
56
 
57
- embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-large')
58
  qna_model = 'gpt-35-turbo'
59
 
60
  chromadb_client = chromadb.PersistentClient(path='./health_policy_db')
 
5
 
6
  from openai import AzureOpenAI
7
 
8
+ from langchain_huggingface import HuggingFaceEmbeddings
9
  from langchain_community.vectorstores import Chroma
10
 
11
  from typing import List
 
54
  api_version = "2024-02-01"
55
  )
56
 
57
+ embedding_model = HuggingFaceEmbeddings(model_name='thenlper/gte-large')
58
  qna_model = 'gpt-35-turbo'
59
 
60
  chromadb_client = chromadb.PersistentClient(path='./health_policy_db')