Spaces:

tasmimulhuda
/

fastapi-rag-qa

Runtime error

tasmimulhuda commited on Nov 2, 2024

Commit

91e1c5b

1 Parent(s): ba44bc5

dimensions issue

Files changed (1) hide show

app/rag_pipeline/model_initializer.py CHANGED Viewed

@@ -27,7 +27,7 @@ def initialize_models(openai_api_key=None,model_id=None, model_basename=None):
             llm_model = ChatOpenAI(api_key=openai_api_key)
             logger.info("Using OpenAI models.")
         else:
-            embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", # all-mpnet-base-v2
                                                     model_kwargs={'device': 'cpu'},
                                                     encode_kwargs={'normalize_embeddings': False},
                                                     cache_folder = CACHE_DIR

             llm_model = ChatOpenAI(api_key=openai_api_key)
             logger.info("Using OpenAI models.")
         else:
+            embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2", # all-MiniLM-L6-v2
                                                     model_kwargs={'device': 'cpu'},
                                                     encode_kwargs={'normalize_embeddings': False},
                                                     cache_folder = CACHE_DIR