1MR commited on
Commit
e73a2ca
·
verified ·
1 Parent(s): 971e413

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -63,10 +63,11 @@ def get_text_chunks(documents):
63
 
64
 
65
  def get_vectorstore(text_chunks):
66
- embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
67
  vectorstore = FAISS.from_documents(text_chunks, embeddings)
68
  return vectorstore
69
-
 
70
 
71
  def get_conversation_chain(vectorstore, tokenH):
72
  if not tokenH:
@@ -80,7 +81,7 @@ def get_conversation_chain(vectorstore, tokenH):
80
  def generate_response(messages):
81
  try:
82
  completion = client.chat.completions.create(
83
- model="microsoft/Phi-3.5-mini-instruct",
84
  messages=messages,
85
  max_tokens=500
86
  )
 
63
 
64
 
65
  def get_vectorstore(text_chunks):
66
+ embeddings = HuggingFaceEmbeddings(model_name="WhereIsAI/UAE-Large-V1")
67
  vectorstore = FAISS.from_documents(text_chunks, embeddings)
68
  return vectorstore
69
+ #sentence-transformers/all-MiniLM-L6-v2
70
+ #HuggingFaceH4/zephyr-7b-alpha
71
 
72
  def get_conversation_chain(vectorstore, tokenH):
73
  if not tokenH:
 
81
  def generate_response(messages):
82
  try:
83
  completion = client.chat.completions.create(
84
+ model="HuggingFaceH4/zephyr-7b-alphat",
85
  messages=messages,
86
  max_tokens=500
87
  )