Shreyas094 commited on
Commit
9bac56d
·
verified ·
1 Parent(s): bd7a464

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def load_document(file: NamedTemporaryFile, parser: str = "llamaparse") -> List[
65
  raise ValueError("Invalid parser specified. Use 'pypdf' or 'llamaparse'.")
66
 
67
  def get_embeddings():
68
- return HuggingFaceEmbeddings(model_name="sentence-transformers/stsb-roberta-large")
69
 
70
  # Add this at the beginning of your script, after imports
71
  DOCUMENTS_FILE = "uploaded_documents.json"
 
65
  raise ValueError("Invalid parser specified. Use 'pypdf' or 'llamaparse'.")
66
 
67
  def get_embeddings():
68
+ return HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
69
 
70
  # Add this at the beginning of your script, after imports
71
  DOCUMENTS_FILE = "uploaded_documents.json"