Update app.py

app.py

@@ -74,7 +74,7 @@ if not HF_TOKEN:
     exit(1)
 
 # HuggingFace Embeddings
-embeddings = HuggingFaceEmbeddings(model_name="BAAI/bge-large-en-v1.5")
+embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
 # Qdrant Client Setup
 try:
@@ -95,7 +95,7 @@ try:
     client.create_collection(
         collection_name=collection_name,
         vectors_config=models.VectorParams(
-            size=768,  # GTE-large embedding size
+            size=384,  # GTE-large embedding size
             distance=models.Distance.COSINE
         )
     )
@@ -115,16 +115,16 @@ db = Qdrant(
 )
 
 # Create retriever
-# retriever = db.as_retriever(
-#     search_type="similarity",
-#     search_kwargs={"k": 5}
-# )
-
 retriever = db.as_retriever(
-    search_type="mmr",
-    search_kwargs={"k": 5, "fetch_k": 10, "lambda_mult": 0.5}
+    search_type="similarity",
+    search_kwargs={"k": 5}
 )
 
+# retriever = db.as_retriever(
+#     search_type="mmr",
+#     search_kwargs={"k": 5, "fetch_k": 10, "lambda_mult": 0.5}
+# )
+