Update app.py

app.py

@@ -7,24 +7,13 @@ from transformers import AutoTokenizer
 import pickle
 import os
 
-from glob import glob
+github_url = "https://github.com/TheMITTech/shakespeare"
 
-files = glob("./shakespeare/**/*.html")
+with open('shakespeare.pkl', 'wb') as fp:
+    pickle.dump(github_url, fp)
 
-import shutil
-import os
-
-os.mkdir('./data')
-destination_folder = './data/'
-
-for html_file in files:
-  shutil.move(html_file, destination_folder + html_file.split("/")[-1])
-
-from langchain.document_loaders import BSHTMLLoader, DirectoryLoader
-
-bshtml_dir_loader = DirectoryLoader('./data/', loader_cls=BSHTMLLoader)
-
-data = bshtml_dir_loader.load()
+with open('shakespeare.pkl', 'rb') as fp:
+    data = pickle.load(fp)
 
 bloomz_tokenizer = AutoTokenizer.from_pretrained('bigscience/bloomz-1b7')