Daemontatox commited on
Commit
41bfdd9
·
verified ·
1 Parent(s): 18c782e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,7 +57,7 @@ CHUTES_KEY = os.getenv("CHUTES_KEY")
57
  # GEMINI=os.getenv("GEMINI_API_KEY")
58
 
59
 
60
- embeddings = HuggingFaceEmbeddings(model_name="ibm-granite/granite-embedding-small-english-r2")
61
 
62
  try:
63
  client = QdrantClient(
@@ -93,7 +93,7 @@ db = Qdrant(
93
 
94
  retriever = db.as_retriever(
95
  search_type="similarity",
96
- search_kwargs={"k": 5}
97
  )
98
 
99
  # Create a new collection for logs
 
57
  # GEMINI=os.getenv("GEMINI_API_KEY")
58
 
59
 
60
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
61
 
62
  try:
63
  client = QdrantClient(
 
93
 
94
  retriever = db.as_retriever(
95
  search_type="similarity",
96
+ search_kwargs={"k": 10}
97
  )
98
 
99
  # Create a new collection for logs