Update app.py

app.py

@@ -13,13 +13,13 @@ from sentence_transformers import SentenceTransformer
 
 embeddings = OpenAIEmbeddings()
 
-embedding_function = SentenceTransformer("BAAI/bge-large-en-v1.5")
+embedder = SentenceTransformer('all-mpnet-base-v2')
 
 # Set the OpenAI API key
 openai.api_key = os.getenv("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")
 
 # Load the FAISS index using LangChain's FAISS implementation
-db = FAISS.load_local("Faiss_indexes", embedding_function, allow_dangerous_deserialization=True)
+db = FAISS.load_local("Faiss_indexes", embedder, allow_dangerous_deserialization=True)
 parser = StrOutputParser()
 
 # Load your data (e.g., a DataFrame)
@@ -27,7 +27,7 @@ df = pd.read_pickle('df_news (1).pkl')
 
 # Search function to retrieve relevant documents
 def search(query):
-    query_embedding = embeddings.embed_query(query).reshape(1, -1).astype('float32')
+    query_embedding = embedder.embed_query(query).reshape(1, -1).astype('float32')
     D, I = db.similarity_search_with_score(query_embedding, k=10)
     results = []
     for idx in I[0]: