Update app.py

app.py

@@ -138,12 +138,12 @@ class ChatModel:
         try:
             # Load model and vector store
             self.load_model(model_name)
-            vectorstore = self.load_vector_store(vector_store_name)
+            self.load_vector_store(vector_store_name)
             config = MODEL_CONFIG[model_name]
             
             # Retrieve relevant context
             logger.info(f"Retrieving context for query: {message}")
-            docs = vectorstore.similarity_search(message, k=3)
+            docs = self.current_vectorstore.similarity_search(message, k=3)
             
             # Log retrieved documents for debugging
             for i, doc in enumerate(docs):