Update app.py

app.py

@@ -33,7 +33,7 @@ class ChatModel:
     def __init__(self):
         self.models = {}
         self.tokenizers = {}
-        self.vectorstore = None
+        self.vectorstore = {}
 
     def load_model(self, model_name):
         if model_name not in self.models:
@@ -148,6 +148,7 @@ class ChatModel:
 model_handler = ChatModel()
 
 def chat(message, history, model_choice, vector_store_choice):
+    print("vector_store_choice: ", vector_store_choice)
     try:
         response, response_time, token_speed = model_handler.generate(message, model_choice, vector_store_choice, history)
         formatted_response = f"{response}\n\n⏱️ Response Time: {response_time:.2f}s | 🚀 Speed: {token_speed:.2f} tokens/s"