Update webchat.py

webchat.py

@@ -79,10 +79,17 @@ def get_model_test(model_type, max_tokens, min_tokens, decoding, temperature):
 
     return model
 
-
+import os
+# Get the current working directory
+current_dir = os.getcwd()
+cache_dir = os.path.join(current_dir, ".cache")
+if not os.path.exists(cache_dir):
+    os.makedirs(cache_dir)
 # Embedding function
 class MiniLML6V2EmbeddingFunction(EmbeddingFunction):
-    MODEL = SentenceTransformer('all-MiniLM-L6-v2')
+    #MODEL = SentenceTransformer('all-MiniLM-L6-v2')
+    MODEL = SentenceTransformer('all-MiniLM-L6-v2', cache_dir=cache_dir)
+
 
     def __call__(self, texts):
         return MiniLML6V2EmbeddingFunction.MODEL.encode(texts).tolist()