Shreyas094 commited on
Commit
f23ec49
·
verified ·
1 Parent(s): f7d0dd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def load_document(file: NamedTemporaryFile, parser: str = "llamaparse") -> List[
67
  raise ValueError("Invalid parser specified. Use 'pypdf' or 'llamaparse'.")
68
 
69
  def get_embeddings():
70
- return HuggingFaceEmbeddings(model_name="dunzhang/stella_en_1.5B_v5")
71
 
72
  # Add this at the beginning of your script, after imports
73
  DOCUMENTS_FILE = "uploaded_documents.json"
 
67
  raise ValueError("Invalid parser specified. Use 'pypdf' or 'llamaparse'.")
68
 
69
  def get_embeddings():
70
+ return HuggingFaceEmbeddings(model_name="McGill-NLP/LLM2Vec-Sheared-LLaMA-mntp-supervised")
71
 
72
  # Add this at the beginning of your script, after imports
73
  DOCUMENTS_FILE = "uploaded_documents.json"