Update app.py

app.py

@@ -28,11 +28,17 @@ BASE_DIR = os.getcwd()
 #replicate api token
 os.environ["REPLICATE_API_TOKEN"] = "r8_30xo4KYovs74WNJiDFmZFENUcoXUBJa1B0nat"
 
+cache_directory = '/cache'
+
+# Create the cache directory if it doesn't exist
+os.makedirs(cache_directory, exist_ok=True)
+
+
 #intialize web search wrapper
 search = DuckDuckGoSearchRun()
 
 #intialize emebding modelr
-embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
+embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',cache_dir=cache_directory)
 
 #milvus database connection
 collection_name = 'LangChainCollection'