Updated embeddings

app.py

@@ -13,12 +13,6 @@ from azure.storage.fileshare import ShareServiceClient
 import re
 import json
 
-
-# Langchain
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.schema import AIMessage, HumanMessage
-from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
-
 # ClimateQ&A imports
 from climateqa.engine.llm import get_llm
 # from climateqa.chains import load_qa_chain_with_docs,load_qa_chain_with_text
@@ -46,6 +40,8 @@ theme = gr.themes.Base(
     font=[gr.themes.GoogleFont("Poppins"), "ui-sans-serif", "system-ui", "sans-serif"],
 )
 
+print("1")
+
 
 
 init_prompt = ""
@@ -91,7 +87,10 @@ def parse_output_llm_with_sources(output):
 
 
 # Create embeddings function and LLM
+print("1")
 embeddings_function = get_embeddings_function()
+print("1")
+
 
 # Create vectorstore and retriever
 vectorstore = get_pinecone_vectorstore(embeddings_function)

climateqa/engine/embeddings.py

@@ -11,6 +11,7 @@ def get_embeddings_function(version = "v1.2"):
     
         model_name = "BAAI/bge-base-en-v1.5"
         encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
+        print("Loading embeddings model: ", model_name)
         embeddings_function = HuggingFaceBgeEmbeddings(
             model_name=model_name,
             encode_kwargs=encode_kwargs,