Update app.py

app.py

@@ -6,6 +6,14 @@ from langchain.chains import RetrievalQA
 from transformers import AutoTokenizer
 import pickle
 import os
+from langchain.document_loaders import BSHTMLLoader, DirectoryLoader
+
+bshtml_dir_loader = DirectoryLoader('./data/', loader_cls = BSHTMLLoader)
+
+data = bshtml_dir_loader.load()
+
+with open("shakespeare.pkl", "wb") as fp:
+  pickle.dump(data, fp)
 
 with open('shakespeare.pkl', 'rb') as fp:
     data = pickle.load(fp)